8686 # Ensure `databricks-connect` and `databricks-agents` are not installed
8787 pip uninstall -y databricks-connect databricks-agents
8888 pip install --no-deps --force-reinstall pyspark
89+ # install uv==0.8.18 to meet tool.uv requirements in pyproject.toml
90+ pip install uv==0.8.18
8991 - uses : ./.github/actions/show-versions
9092 - uses : ./.github/actions/pipdeptree
9193 - name : Import check
@@ -219,8 +221,6 @@ jobs:
219221 tests/tracking/fluent/test_fluent_autolog.py \
220222 tests/autologging
221223
222- # It takes 9 ~ 10 minutes to run tests in `tests/models`. To make CI finish faster,
223- # run these tests in a separate job.
224224 models :
225225 if : github.event_name != 'pull_request' || github.event.pull_request.draft == false
226226 runs-on : ubuntu-latest
@@ -250,10 +250,6 @@ jobs:
250250 run : |
251251 pytest --splits=${{ matrix.splits }} --group=${{ matrix.group }} tests/models
252252
253- # NOTE: numpy is pinned in this suite due to its heavy reliance on shap, which internally uses
254- # references to the now fully deprecated (as of 1.24.x) numpy types (i.e., np.bool).
255- # When the shap cross version tests are passing in a new release version of shap, this pin should
256- # be removed.
257253 evaluate :
258254 if : github.event_name != 'pull_request' || github.event.pull_request.draft == false
259255 runs-on : ubuntu-latest
@@ -274,16 +270,18 @@ jobs:
274270 - uses : ./.github/actions/setup-java
275271 - name : Install dependencies
276272 run : |
277- source ./dev/install-common-deps.sh
278- pip install pyspark torch transformers langchain langchain-experimental '.[genai]' 'shap<0.47.0' lightgbm xgboost
273+ uv sync --extra extras --extra jobs --extra genai
274+ uv pip install \
275+ -r requirements/test-requirements.txt \
276+ torch transformers 'pyspark<3.5.6' langchain langchain-experimental 'shap<0.47.0' lightgbm xgboost
279277 - uses : ./.github/actions/show-versions
280278 - uses : ./.github/actions/pipdeptree
281279 - name : Run tests
282280 run : |
283- pytest --splits=${{ matrix.splits }} --group=${{ matrix.group }} tests/evaluate --ignore=tests/evaluate/test_default_evaluator_delta.py
281+ uv run pytest --splits=${{ matrix.splits }} --group=${{ matrix.group }} tests/evaluate --ignore=tests/evaluate/test_default_evaluator_delta.py
284282 - name : Run tests with delta
285283 run : |
286- pytest tests/evaluate/test_default_evaluator_delta.py
284+ uv run pytest tests/evaluate/test_default_evaluator_delta.py
287285
288286 genai :
289287 if : github.event_name != 'pull_request' || github.event.pull_request.draft == false
0 commit comments