Skip to content

Commit faaa9f4

Browse files
committed
fix bug when NVIDIA_INTERNAL not in environ
1 parent 4a6ad51 commit faaa9f4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

runx/runx.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,7 @@ def construct_cmd(cmd, hparams, resources, job_name, logdir):
104104
######################################################################
105105
# You may wish to customize this function for your own needs
106106
######################################################################
107-
if os.environ['NVIDIA_INTERNAL']:
107+
if os.environ.get('NVIDIA_INTERNAL'):
108108
cmd += '--name {} '.format(job_name)
109109
if 'submit_job' in cmd:
110110
cmd += '--cd_to_logdir '

0 commit comments

Comments
 (0)