Running opt-baselines requires os.environ["USER"]

[Leli1024]

🐛 Bug

When I'm trying to load the model in some way, I keep getting an error that the USER environment variable is missing, I have attempted to fix this by overriding the variable manually, however I still get an error raise RuntimeError("Where are you running this?! Check DATA_LOCATIONS.") RuntimeError: Where are you running this?! Check DATA_LOCATIONS.

To Reproduce

Steps to reproduce the behavior (always include the command you ran):

1. Run opt-baselines -n 2 -g 8 -p test_v0 --model-size 1.3b --azure --checkpoints-dir "model_ckpts" --no-save-dir # Remove this if you want to print out full save-dir path
2. See error Traceback (most recent call last): File "/home/studio-lab-user/.conda/envs/studiolab/bin/opt-baselines", line 33, in <module> sys.exit(load_entry_point('metaseq', 'console_scripts', 'opt-baselines')()) File "/home/studio-lab-user/metaseq/metaseq/launcher/opt_baselines.py", line 313, in cli_main sweep_main( File "/home/studio-lab-user/metaseq/metaseq/launcher/sweep.py", line 374, in main args = _get_args(add_extra_options_func, scheduler_args) File "/home/studio-lab-user/metaseq/metaseq/launcher/sweep.py", line 238, in _get_args _modify_arg_defaults_based_on_env(env, args) File "/home/studio-lab-user/metaseq/metaseq/launcher/sweep.py", line 263, in _modify_arg_defaults_based_on_env os.environ["USER"], File "/home/studio-lab-user/.conda/envs/studiolab/lib/python3.9/os.py", line 679, in __getitem__ raise KeyError(key) from None KeyError: 'USER'

Environment

• python and torch: as specified within the setup readme
• OS (e.g., Linux, Windows, MacOS): linux (AWS instance)
• How you installed metaseq (pip, source): source

[stephenroller]

Guess we need to change that to os.getuser()?

[Leli1024]

I tried this but apparently it gives a function doesn't exist error, I'm assuming it's a Linux specific issue maybe?

[stephenroller]

https://stackoverflow.com/a/842096 my memory was slightly off