@@ -90,14 +90,15 @@ jobs:
9090 - name : Import check
9191 run : |
9292 source .venv/bin/activate
93- python tests/check_mlflow_lazily_imports_ml_packages.py
93+ # `-I` is used to avoid importing modules from user-specific site-packages
94+ # that might conflict with the built-in modules (e.g. `types`).
95+ python -I tests/check_mlflow_lazily_imports_ml_packages.py
9496 - name : Run tests
9597 run : |
9698 source .venv/bin/activate
9799 source dev/setup-ssh.sh
98100 pytest --splits=${{ matrix.splits }} --group=${{ matrix.group }} --quiet --requires-ssh \
99- --ignore-flavors --ignore=tests/examples --ignore=tests/recipes --ignore=tests/evaluate \
100- --ignore=tests/deployments/server --ignore=tests/pyspark/optuna \
101+ --ignore-flavors --ignore=tests/examples --ignore=tests/evaluate \
101102 --ignore tests/genai tests
102103
103104 py310 :
@@ -332,10 +333,16 @@ jobs:
332333 - name : Install dependencies
333334 run : |
334335 source ./dev/install-common-deps.sh
336+ # GenAI evaluation tests require databricks-agents that only work with Python >= 3.10
337+ pip install 'databricks-agents<0.22.0' openai
338+ - uses : ./.github/actions/show-versions
335339 - name : Run GenAI Tests
336340 run : |
337341 # GenAI evaluation tests require databricks-agents that only work with Python 3.10~
338- pip install databricks-agents openai
342+ pip install openai
343+ # Install databricks-agents from a pre-release wheel because the latest released version
344+ # is not compatible with MLflow master. Using --no-deps because it pings mlflow to 3.0.0rc2.
345+ pip install https://ml-team-public-read.s3.us-west-2.amazonaws.com/wheels/rag-studio/dev/samraj.moorjani/databricks_agents-1.0.0rc1.dev0-py3-none-any.whl --no-deps
339346 pytest tests/genai
340347
341348 optuna :
@@ -481,5 +488,5 @@ jobs:
481488 export PATH=$PATH:$HADOOP_HOME/bin
482489 # Run Windows tests
483490 pytest --splits=${{ matrix.splits }} --group=${{ matrix.group }} \
484- --ignore-flavors --ignore=tests/projects --ignore=tests/examples --ignore=tests/recipes --ignore=tests/ evaluate --ignore tests/deployments/server --ignore=tests/optuna \
491+ --ignore-flavors --ignore=tests/projects --ignore=tests/examples --ignore=tests/evaluate --ignore=tests/optuna \
485492 --ignore=tests/genai tests
0 commit comments