Issue with ns-install-cli -- 'contextmanager' is not defined

[benijohn]

I am not sure if this is a bug or user error. I am trying to install the "adam/terrain" branch of nerfstudio. Everything was good up until I rand the line

ns-install-cli

I recieved the following error:

`ns-install-cli [10:49:03] 🔍 Detected conda environment /home/benjamin/miniconda3/envs/nerfstudio! install.py:363 ✔ Nothing to do for install.py:138 /home/benjamin/workspace/projects/nerfstudio/nerfstudio/scripts/completions/bash/_ns-dev-test.
✔ Nothing to do for install.py:138 /home/benjamin/workspace/projects/nerfstudio/nerfstudio/scripts/completions/bash/_ns-install-c
li.
✔ Nothing to do for install.py:138 /home/benjamin/workspace/projects/nerfstudio/nerfstudio/scripts/completions/zsh/_ns-dev-test.
✔ Nothing to do for install.py:138 /home/benjamin/workspace/projects/nerfstudio/nerfstudio/scripts/completions/zsh/_ns-install-cl
i.
[10:49:05] ✔ Nothing to do for install.py:138 /home/benjamin/workspace/projects/nerfstudio/nerfstudio/scripts/completions/zsh/_ns-dev-sync-v
iser-message-defs.
✔ Nothing to do for install.py:138 /home/benjamin/workspace/projects/nerfstudio/nerfstudio/scripts/completions/bash/_ns-dev-sync-
viser-message-defs.
✔ Nothing to do for install.py:138 /home/benjamin/workspace/projects/nerfstudio/nerfstudio/scripts/completions/bash/_ns-download-
data.
✔ Nothing to do for install.py:138 /home/benjamin/workspace/projects/nerfstudio/nerfstudio/scripts/completions/zsh/_ns-download-d
ata.
[10:49:07] ✔ Nothing to do for install.py:138 /home/benjamin/workspace/projects/nerfstudio/nerfstudio/scripts/completions/zsh/_ns-process-da
ta.
✔ Nothing to do for install.py:138 /home/benjamin/workspace/projects/nerfstudio/nerfstudio/scripts/completions/bash/_ns-process-d
ata.
[10:49:09] ✔ Wrote new completion to install.py:133 /home/benjamin/workspace/projects/nerfstudio/nerfstudio/scripts/completions/bash/_ns-eval!
✔ Wrote new completion to install.py:133 /home/benjamin/workspace/projects/nerfstudio/nerfstudio/scripts/completions/zsh/_ns-viewer!
✔ Wrote new completion to install.py:133 /home/benjamin/workspace/projects/nerfstudio/nerfstudio/scripts/completions/bash/_ns-viewer!
✔ Wrote new completion to install.py:133 /home/benjamin/workspace/projects/nerfstudio/nerfstudio/scripts/completions/zsh/_ns-eval!
❌ Completion script generation failed: ['ns-render', '--tyro-print-completion', 'zsh'] install.py:123 Traceback (most recent call last): install.py:127 File "/home/benjamin/miniconda3/envs/nerfstudio/bin/ns-render", line 5, in
from nerfstudio.scripts.render import entrypoint
File "/home/benjamin/workspace/projects/nerfstudio/nerfstudio/scripts/render.py", line 607,
in
@contextmanager
NameError: name 'contextmanager' is not defined

[10:49:10] ❌ Completion script generation failed: ['ns-render', '--tyro-print-completion', 'bash'] install.py:123 Traceback (most recent call last): install.py:127 File "/home/benjamin/miniconda3/envs/nerfstudio/bin/ns-render", line 5, in
from nerfstudio.scripts.render import entrypoint
File "/home/benjamin/workspace/projects/nerfstudio/nerfstudio/scripts/render.py", line 607,
in
@contextmanager
NameError: name 'contextmanager' is not defined

       ✔ Wrote new completion to                                                                      install.py:133
       /home/benjamin/workspace/projects/nerfstudio/nerfstudio/scripts/completions/bash/_ns-train!                  
       ✔ Wrote new completion to                                                                      install.py:133
       /home/benjamin/workspace/projects/nerfstudio/nerfstudio/scripts/completions/zsh/_ns-train!                   

[10:49:12] ✔ Wrote new completion to install.py:133 /home/benjamin/workspace/projects/nerfstudio/nerfstudio/scripts/completions/zsh/_ns-export!
✔ Wrote new completion to install.py:133 /home/benjamin/workspace/projects/nerfstudio/nerfstudio/scripts/completions/bash/_ns-export!
Traceback (most recent call last): File "/home/benjamin/miniconda3/envs/nerfstudio/bin/ns-install-cli", line 8, in sys.exit(entrypoint()) File "/home/benjamin/workspace/projects/nerfstudio/nerfstudio/scripts/completions/install.py", line 411, in entrypoint tyro.cli(main, description=doc) File "/home/benjamin/miniconda3/envs/nerfstudio/lib/python3.9/site-packages/tyro/_cli.py", line 229, in cli return run_with_args_from_cli() File "/home/benjamin/workspace/projects/nerfstudio/nerfstudio/scripts/completions/install.py", line 392, in main _generate_completions_files(completions_dir, scripts_dir, shells_supported, shells_found) File "/home/benjamin/workspace/projects/nerfstudio/nerfstudio/scripts/completions/install.py", line 323, in _generate_completions_files completion_paths = list( File "/home/benjamin/miniconda3/envs/nerfstudio/lib/python3.9/concurrent/futures/_base.py", line 609, in result_iterator yield fs.pop().result() File "/home/benjamin/miniconda3/envs/nerfstudio/lib/python3.9/concurrent/futures/_base.py", line 439, in result return self.__get_result() File "/home/benjamin/miniconda3/envs/nerfstudio/lib/python3.9/concurrent/futures/_base.py", line 391, in __get_result raise self._exception File "/home/benjamin/miniconda3/envs/nerfstudio/lib/python3.9/concurrent/futures/thread.py", line 58, in run result = self.fn(*self.args, **self.kwargs) File "/home/benjamin/workspace/projects/nerfstudio/nerfstudio/scripts/completions/install.py", line 325, in lambda path_or_entrypoint_and_shell: _generate_completion( File "/home/benjamin/workspace/projects/nerfstudio/nerfstudio/scripts/completions/install.py", line 128, in _generate_completion raise e File "/home/benjamin/workspace/projects/nerfstudio/nerfstudio/scripts/completions/install.py", line 115, in _generate_completion new = subprocess.run( File "/home/benjamin/miniconda3/envs/nerfstudio/lib/python3.9/subprocess.py", line 528, in run raise CalledProcessError(retcode, process.args, subprocess.CalledProcessError: Command '['ns-render', '--tyro-print-completion', 'zsh']' returned non-zero exit status 1. ` I am not sure why this is happening. Am I missing some dependency?

I am on Ubuntu 20.04, Python 3.9, Cuda 11-8, Pytorch 2.12

Please let me know if you need any other information.

Cheers, Benjamin