Bug-fix visualize reconstruction

Summary:
Addresses the following issue:
https://github.com/facebookresearch/pytorch3d/issues/1345#issuecomment-1272881244

I.e., when installed from conda, `pytorch3d_implicitron_visualizer` crashes since it invokes `main()` while `main` requires a single positional arg `argv`.

Reviewed By: shapovalov

Differential Revision: D41533497

fbshipit-source-id: e53a923eb8b2f0f9c0e92e9c0866d9cb310c4799
This commit is contained in:
David Novotny 2022-11-25 10:30:01 -08:00 committed by Facebook GitHub Bot
parent 60ab1cdb72
commit c3a6ab02da

View File

@ -141,7 +141,7 @@ def _get_config_from_experiment_directory(experiment_directory) -> DictConfig:
return OmegaConf.merge(get_default_args(Experiment), config)
def main(argv) -> None:
def main(argv=sys.argv) -> None:
# automatically parses arguments of visualize_reconstruction
cfg = OmegaConf.create(get_default_args(visualize_reconstruction))
cfg.update(OmegaConf.from_cli(argv))
@ -150,4 +150,4 @@ def main(argv) -> None:
if __name__ == "__main__":
main(sys.argv)
main()