Skip to content

Commit 7502837

Browse files
committed
chore: restrict modelopt installation to certain tests
1 parent d03d308 commit 7502837

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

.github/workflows/build-test-linux.yml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -138,6 +138,7 @@ jobs:
138138
pushd .
139139
cd tests/py
140140
python -m pip install -r requirements.txt
141+
python -m pip install nvidia-modelopt[all]; python_version < "3.13" --extra-index-url https://pypi.nvidia.com
141142
cd dynamo
142143
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dynamo_converters_test_results.xml -n 4 conversion/
143144
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dynamo_converters_test_results.xml automatic_plugin/test_automatic_plugin.py
@@ -172,6 +173,7 @@ jobs:
172173
pushd .
173174
cd tests/py
174175
python -m pip install -r requirements.txt
176+
python -m pip install nvidia-modelopt[all]; python_version < "3.13" --extra-index-url https://pypi.nvidia.com
175177
cd dynamo
176178
python -m pytest -ra --junitxml=${RUNNER_TEST_RESULTS_DIR}/dyn_models_export.xml --ir dynamo models/
177179
popd

tests/py/requirements.txt

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,5 +10,4 @@ pyyaml
1010
timm>=1.0.3
1111
flashinfer-python; python_version < "3.13"
1212
transformers==4.49.0
13-
nvidia-modelopt[all]; python_version < "3.13"
1413
--extra-index-url https://pypi.nvidia.com

0 commit comments

Comments
 (0)