Skip to content

Commit

Permalink
fix: Preempt torch package override via timm in nox session
Browse files Browse the repository at this point in the history
Signed-off-by: Anurag Dixit <a.dixit91@gmail.com>
  • Loading branch information
andi4191 committed May 2, 2022
1 parent 1c294fa commit 8964d1b
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion noxfile.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,7 @@ def download_models(session, use_host_env=False):
if use_host_env:
session.run_always('python', 'hub.py', env={'PYTHONPATH': PYT_PATH})
else:
session.install("-r", os.path.join(TOP_DIR, "py", "requirements.txt"))
session.run_always('python', 'hub.py')

def install_torch_trt(session):
Expand Down Expand Up @@ -334,4 +335,4 @@ def download_test_models(session):
@nox.session(python=SUPPORTED_PYTHON_VERSIONS, reuse_venv=True)
def download_test_models_host_deps(session):
"""Grab all the models needed for testing using host dependencies"""
download_models(session, use_host_env=True)
download_models(session, use_host_env=True)

0 comments on commit 8964d1b

Please # to comment.