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