Skip to content

Commit 66e695f

Browse files
committed
fixed so you dont need NVIDIA_INTERNAL defined in environ
1 parent 8b20f0f commit 66e695f

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
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 'NVIDIA_INTERNAL' in os.environ and os.environ['NVIDIA_INTERNAL']:
108108
cmd += '--name {} '.format(job_name)
109109
if 'submit_job' in cmd:
110110
cmd += '--cd_to_logdir '

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
parent = Path(__file__).resolve().parent
1111
setup(
1212
name="runx",
13-
version="0.0.2",
13+
version="0.0.3",
1414
author="Andrew Tao",
1515
author_email="[email protected]",
1616
description="runx - experiment manager for machine learning research",

0 commit comments

Comments
 (0)