Skip to content

Commit ba4b103

Browse files
authored
Merge pull request #1016 from NVIDIA/anuragd/nox_fix_torch_pkg_override
fix: Preempt torch package override via timm in nox session
2 parents 1c294fa + 8964d1b commit ba4b103

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

noxfile.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@ def download_models(session, use_host_env=False):
2727
if use_host_env:
2828
session.run_always('python', 'hub.py', env={'PYTHONPATH': PYT_PATH})
2929
else:
30+
session.install("-r", os.path.join(TOP_DIR, "py", "requirements.txt"))
3031
session.run_always('python', 'hub.py')
3132

3233
def install_torch_trt(session):
@@ -334,4 +335,4 @@ def download_test_models(session):
334335
@nox.session(python=SUPPORTED_PYTHON_VERSIONS, reuse_venv=True)
335336
def download_test_models_host_deps(session):
336337
"""Grab all the models needed for testing using host dependencies"""
337-
download_models(session, use_host_env=True)
338+
download_models(session, use_host_env=True)

0 commit comments

Comments
 (0)