@@ -102,13 +102,6 @@ jobs:
102102 --ignore-flavors --ignore=tests/examples --ignore=tests/evaluate \
103103 --ignore tests/genai tests
104104
105- # TODO: Remove this step once support for pydantic v1 is dropped
106- # - name: Run gateway tests with pydantic v1
107- # run: |
108- # source .venv/bin/activate
109- # pip install 'pydantic<2.0'
110- # pytest --splits=${{ matrix.splits }} --group=${{ matrix.group }} tests/gateway
111-
112105 - name : Run databricks-connect related tests
113106 run : |
114107 # this needs to be run in a separate job because installing databricks-connect could break other
@@ -180,6 +173,8 @@ jobs:
180173 - uses : ./.github/actions/untracked
181174 - uses : ./.github/actions/setup-python
182175 - uses : ./.github/actions/setup-java
176+ with :
177+ java-version : 11
183178 - name : Install dependencies
184179 run : |
185180 source ./dev/install-common-deps.sh
@@ -243,7 +238,7 @@ jobs:
243238 uv sync --extra mlserver
244239 uv pip install \
245240 -r requirements/test-requirements.txt \
246- ' pyspark<3.5.6' langchain langchain-community
241+ pyspark langchain langchain-community
247242 - uses : ./.github/actions/show-versions
248243 - uses : ./.github/actions/pipdeptree
249244 - name : Run tests
@@ -273,7 +268,7 @@ jobs:
273268 uv sync --extra extras --extra jobs --extra genai
274269 uv pip install \
275270 -r requirements/test-requirements.txt \
276- torch transformers ' pyspark<3.5.6' langchain langchain-experimental 'shap<0.47.0' lightgbm xgboost
271+ torch transformers pyspark langchain langchain-experimental 'shap<0.47.0' lightgbm xgboost
277272 - uses : ./.github/actions/show-versions
278273 - uses : ./.github/actions/pipdeptree
279274 - name : Run tests
@@ -343,7 +338,7 @@ jobs:
343338 uv sync --extra extras --extra gateway --extra jobs
344339 uv pip install \
345340 -r requirements/test-requirements.txt \
346- tensorflow 'pyspark[connect]<3.5.6 '
341+ tensorflow 'pyspark[connect]'
347342 - uses : ./.github/actions/show-versions
348343 - uses : ./.github/actions/pipdeptree
349344 - name : Run tests
@@ -353,12 +348,6 @@ jobs:
353348
354349 # test_spark_connect.py fails if it's run with other tests, so run it separately.
355350 uv run --no-sync pytest tests/pyfunc/test_spark_connect.py
356- # TODO: Remove this step once support for pydantic v1 is dropped
357- # - name: Run tests on pydantic v1
358- # run: |
359- # pip install 'pydantic<2'
360- # pytest --splits=${{ matrix.splits }} --group=${{ matrix.group }} --durations=30 \
361- # tests/types tests/pyfunc/test_pyfunc_model_with_type_hints.py tests/utils/test_pydantic_utils.py
362351
363352 windows :
364353 if : github.event_name != 'pull_request' || github.event.pull_request.draft == false
@@ -380,10 +369,10 @@ jobs:
380369 - uses : ./.github/actions/setup-java
381370 - name : Install python dependencies
382371 run : |
383- uv sync --extra extras --extra genai
372+ uv sync --extra extras --extra genai --extra mcp
384373 uv pip install \
385374 -r requirements/test-requirements.txt \
386- ' pyspark<3.5.6' datasets tensorflow torch transformers tf-keras openai \
375+ pyspark datasets tensorflow torch transformers tf-keras openai \
387376 tests/resources/mlflow-test-plugin
388377 - uses : ./.github/actions/show-versions
389378 - uses : ./.github/actions/pipdeptree
0 commit comments