From c75a56543d5a295fd4de0054bcbb987bf616e42b Mon Sep 17 00:00:00 2001 From: Digant Desai Date: Thu, 25 Jun 2026 11:21:44 -0700 Subject: [PATCH] Install yolo26 requirements before portable export in CI test_model.sh exports yolo26 via examples.portable.scripts.export, but yolo26's model imports ultralytics, which was never installed -- the CI job failed with ModuleNotFoundError: No module named 'ultralytics'. Install examples/models/yolo26/requirements.txt before the export, matching the per-model dep handling already used for llava/edsr/mb. torch/torchvision are already installed and satisfy ultralytics, so --upgrade-strategy only-if-needed leaves them untouched and no pytorch extra-index-url is needed. Authored with Claude assistance. --- .ci/scripts/test_model.sh | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/.ci/scripts/test_model.sh b/.ci/scripts/test_model.sh index c056b5ac98a..a0ef093d91a 100755 --- a/.ci/scripts/test_model.sh +++ b/.ci/scripts/test_model.sh @@ -127,6 +127,14 @@ test_model() { return fi + if [[ "${MODEL_NAME}" == "yolo26" ]]; then + # Install yolo26 requirements (ultralytics). torch/torchvision are already + # installed and satisfy ultralytics, so only-if-needed leaves them as-is. + # No pytorch extra-index-url: it is unused here and would broaden pip's + # version resolution across all deps. + "${PYTHON_EXECUTABLE}" -m pip install --upgrade-strategy only-if-needed -r examples/models/yolo26/requirements.txt + fi + # Export a basic .pte and run the model. "${PYTHON_EXECUTABLE}" -m examples.portable.scripts.export --model_name="${MODEL_NAME}" "${STRICT}" run_portable_executor_runner