From 5345b10b6ad4034a8dbe633fce9ce4c509942801 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Fri, 15 May 2026 01:25:23 -0700 Subject: [PATCH] ci: install ipython so transformers.utils.notebook imports cleanly in zoo pytest (#5437) unsloth_zoo's drift-detector tests/test_zoo_source_upstream_refs.py:: test_logging_utils_utils_notebook resolves transformers.utils.notebook, which executes ``import IPython.display as disp`` at module scope. The Core matrix install list did not include IPython, so the import raised ModuleNotFoundError and the test failed with: DRIFT DETECTED: transformers.utils.notebook exists but its imports fail on this install (ModuleNotFoundError: No module named 'IPython') The test message itself states the resolution: "Either install the dep in CI or remove the zoo reference." Installing keeps the upstream-refs detector functional. Add ipython to the matrix install list. --- .github/workflows/consolidated-tests-ci.yml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/.github/workflows/consolidated-tests-ci.yml b/.github/workflows/consolidated-tests-ci.yml index 218d467f04..2b0ce43a7c 100644 --- a/.github/workflows/consolidated-tests-ci.yml +++ b/.github/workflows/consolidated-tests-ci.yml @@ -206,7 +206,8 @@ jobs: 'numpy<3' pytest==9.0.3 pytest-asyncio httpx \ protobuf sentencepiece triton \ psutil packaging tqdm safetensors datasets \ - 'peft>=0.18,<0.20' 'accelerate>=0.34,<2' + 'peft>=0.18,<0.20' 'accelerate>=0.34,<2' \ + ipython # torchvision: unsloth_zoo.vision_utils imports it at module scope. pip install --index-url https://download.pytorch.org/whl/cpu \ 'torch>=2.4,<2.11' 'torchvision<0.26'