|
95 | 95 | source dev/setup-ssh.sh |
96 | 96 | pytest --splits=${{ matrix.splits }} --group=${{ matrix.group }} --quiet --requires-ssh \ |
97 | 97 | --ignore-flavors --ignore=tests/examples --ignore=tests/recipes --ignore=tests/evaluate \ |
98 | | - --ignore tests/deployments/server tests |
| 98 | + --ignore=tests/deployments/server --ignore=tests/pyspark/optuna tests |
99 | 99 |
|
100 | 100 | py310: |
101 | 101 | if: github.event_name != 'pull_request' || github.event.pull_request.draft == false |
@@ -302,6 +302,30 @@ jobs: |
302 | 302 | run: | |
303 | 303 | pytest tests/evaluate/test_default_evaluator_delta.py |
304 | 304 |
|
| 305 | + optuna: |
| 306 | + if: github.event_name != 'pull_request' || github.event.pull_request.draft == false |
| 307 | + runs-on: ubuntu-latest |
| 308 | + timeout-minutes: 90 |
| 309 | + permissions: {} |
| 310 | + strategy: |
| 311 | + fail-fast: false |
| 312 | + steps: |
| 313 | + - uses: actions/checkout@v4 |
| 314 | + with: |
| 315 | + submodules: recursive |
| 316 | + - uses: ./.github/actions/untracked |
| 317 | + - uses: ./.github/actions/setup-python |
| 318 | + - uses: ./.github/actions/setup-pyenv |
| 319 | + - uses: ./.github/actions/setup-java |
| 320 | + - name: Install dependencies |
| 321 | + run: | |
| 322 | + source ./dev/install-common-deps.sh |
| 323 | + pip install 'pyspark>=3.5' 'optuna>=4' . |
| 324 | + - uses: ./.github/actions/show-versions |
| 325 | + - uses: ./.github/actions/pipdeptree |
| 326 | + - name: Run tests |
| 327 | + run: | |
| 328 | + pytest tests/pyspark/optuna/test_storage.py |
305 | 329 | pyfunc: |
306 | 330 | if: github.event_name != 'pull_request' || github.event.pull_request.draft == false |
307 | 331 | runs-on: ubuntu-latest |
@@ -422,7 +446,7 @@ jobs: |
422 | 446 | export PATH=$PATH:$HADOOP_HOME/bin |
423 | 447 | # Run Windows tests |
424 | 448 | pytest --splits=${{ matrix.splits }} --group=${{ matrix.group }} \ |
425 | | - --ignore-flavors --ignore=tests/projects --ignore=tests/examples --ignore=tests/recipes --ignore=tests/evaluate --ignore tests/deployments/server \ |
| 449 | + --ignore-flavors --ignore=tests/projects --ignore=tests/examples --ignore=tests/recipes --ignore=tests/evaluate --ignore tests/deployments/server --ignore=tests/pyspark/optuna \ |
426 | 450 | tests |
427 | 451 | # MLeap is incompatible on Windows with PySpark3.4 release. |
428 | 452 | # Reinstate tests when MLeap has released a fix. [ML-30491] |
|
0 commit comments