6464 strategy :
6565 fail-fast : false
6666 matrix :
67- group : [1, 2]
67+ group : [1, 2, 3 ]
6868 include :
69- - splits : 2
69+ - splits : 3
7070 steps :
7171 - uses : actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
7272 - uses : ./.github/actions/untracked
@@ -98,34 +98,17 @@ jobs:
9898 --ignore-flavors --ignore=tests/examples --ignore=tests/evaluate \
9999 --ignore tests/genai tests
100100
101- py310 :
102- if : github.event_name != 'pull_request' || github.event.pull_request.draft == false
103- runs-on : ubuntu-latest
104- timeout-minutes : 60
105- permissions :
106- contents : read
107- steps :
108- - uses : actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
109- - uses : ./.github/actions/untracked
110- - uses : ./.github/actions/setup-python
111- - uses : ./.github/actions/setup-pyenv
112- - uses : ./.github/actions/setup-java
113- - uses : ./.github/actions/cache-pip
114- - name : Install dependencies
101+ - name : Run gateway tests with pydantic v1
115102 run : |
116- source ./dev/install-common-deps.sh
117- # test telemetry events
118- pip install openai
119- pip install .
120- - uses : ./.github/actions/show-versions
121- - name : Run tests
122- run : |
123- python -c "import sys; assert sys.version_info[:2] == (3, 10), sys.version_info"
124- pytest tests/types/test_type_hints.py tests/pyfunc/test_pyfunc_model_with_type_hints.py tests/telemetry
103+ source .venv/bin/activate
104+ pip install 'pydantic<2.0'
105+ pytest --splits=${{ matrix.splits }} --group=${{ matrix.group }} tests/gateway
106+
125107 - name : Run databricks-connect related tests
126108 run : |
127- # this needs to be run in a separate job because installing databricks-connect could break other
109+ # this needs to be run in a separate job because installing databricks-connect could break other
128110 # tests that uses normal SparkSession instead of remote SparkSession
111+ source .venv/bin/activate
129112 pip install databricks-agents
130113 pytest tests/utils/test_requirements_utils.py::test_infer_pip_requirements_on_databricks_agents
131114
@@ -230,10 +213,8 @@ jobs:
230213 run : |
231214 source .venv/bin/activate
232215 pytest \
233- tests/utils/test_model_utils.py \
234216 tests/tracking/fluent/test_fluent_autolog.py \
235- tests/autologging \
236- tests/server/auth
217+ tests/autologging
237218
238219 # It takes 9 ~ 10 minutes to run tests in `tests/models`. To make CI finish faster,
239220 # run these tests in a separate job.
@@ -337,27 +318,6 @@ jobs:
337318 pip install pyspark
338319 pytest tests/genai/evaluate/test_utils.py
339320
340- optuna :
341- if : github.event_name != 'pull_request' || github.event.pull_request.draft == false
342- runs-on : ubuntu-latest
343- timeout-minutes : 30
344- permissions :
345- contents : read
346- steps :
347- - uses : actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
348- - uses : ./.github/actions/untracked
349- - uses : ./.github/actions/setup-python
350- - uses : ./.github/actions/setup-pyenv
351- - uses : ./.github/actions/setup-java
352- - name : Install dependencies
353- run : |
354- source ./dev/install-common-deps.sh
355- pip install 'pyspark>=3.5' 'optuna>=4' .
356- - uses : ./.github/actions/show-versions
357- - uses : ./.github/actions/pipdeptree
358- - name : Run tests
359- run : |
360- pytest tests/optuna tests/pyspark/optuna
361321 pyfunc :
362322 if : github.event_name != 'pull_request' || github.event.pull_request.draft == false
363323 runs-on : ubuntu-latest
@@ -381,7 +341,6 @@ jobs:
381341 run : |
382342 source ./dev/install-common-deps.sh
383343 pip install tensorflow 'pyspark[connect]'
384- pip install typing_extensions -U
385344 - uses : ./.github/actions/show-versions
386345 - uses : ./.github/actions/pipdeptree
387346 - name : Run tests
@@ -397,26 +356,6 @@ jobs:
397356 pytest --splits=${{ matrix.splits }} --group=${{ matrix.group }} --durations=30 \
398357 tests/types tests/pyfunc/test_pyfunc_model_with_type_hints.py tests/utils/test_pydantic_utils.py
399358
400- sagemaker :
401- if : github.event_name != 'pull_request' || github.event.pull_request.draft == false
402- runs-on : ubuntu-latest
403- timeout-minutes : 120
404- permissions :
405- contents : read
406- steps :
407- - uses : actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
408- - uses : ./.github/actions/untracked
409- - uses : ./.github/actions/setup-python
410- - uses : ./.github/actions/setup-java
411- - name : Install dependencies
412- run : |
413- source ./dev/install-common-deps.sh --ml
414- - uses : ./.github/actions/show-versions
415- - uses : ./.github/actions/pipdeptree
416- - name : Run tests
417- run : |
418- pytest tests/sagemaker
419-
420359 windows :
421360 if : github.event_name != 'pull_request' || github.event.pull_request.draft == false
422361 runs-on : windows-latest
@@ -426,9 +365,9 @@ jobs:
426365 strategy :
427366 fail-fast : false
428367 matrix :
429- group : [1, 2]
368+ group : [1, 2, 3, 4 ]
430369 include :
431- - splits : 2
370+ - splits : 4
432371 steps :
433372 - uses : actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
434373 - uses : ./.github/actions/untracked
@@ -474,9 +413,14 @@ jobs:
474413 export PATH=$PATH:$HADOOP_HOME/bin
475414 # Run Windows tests
476415 pytest --splits=${{ matrix.splits }} --group=${{ matrix.group }} \
477- --ignore-flavors --ignore=tests/projects --ignore=tests/examples --ignore=tests/evaluate --ignore=tests/optuna \
478- --ignore=tests/pyspark/optuna --ignore=tests/genai --ignore=tests/telemetry tests
479- - name : Run telemetry tests
480- run : |
481- source .venv/Scripts/activate
482- pytest --splits=${{ matrix.splits }} --group=${{ matrix.group }} tests/telemetry
416+ --ignore-flavors \
417+ --ignore=tests/projects \
418+ --ignore=tests/examples \
419+ --ignore=tests/evaluate \
420+ --ignore=tests/optuna \
421+ --ignore=tests/pyspark/optuna \
422+ --ignore=tests/genai \
423+ --ignore=tests/sagemaker \
424+ --ignore=tests/gateway \
425+ --ignore=tests/server/auth \
426+ tests
0 commit comments