MLflow Slow Tests #556
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # A daily job to run slow tests with MLFLOW_RUN_SLOW_TESTS environment variable set to true. | |
| name: MLflow Slow Tests | |
| on: | |
| pull_request: | |
| paths: | |
| # Run this workflow in PR when relevant files change | |
| - ".github/workflows/slow-tests.yml" | |
| - "tests/pyfunc/docker/**" | |
| workflow_dispatch: | |
| inputs: | |
| repository: | |
| description: > | |
| [Optional] Repository name with owner. For example, mlflow/mlflow. | |
| Defaults to the repository that triggered a workflow. | |
| required: false | |
| default: "" | |
| ref: | |
| description: > | |
| [Optional] The branch, tag or SHA to checkout. When checking out the repository that | |
| triggered a workflow, this defaults to the reference or SHA for that event. Otherwise, | |
| uses the default branch. | |
| required: false | |
| default: "" | |
| schedule: | |
| # Run this workflow daily at 13:00 UTC | |
| - cron: "0 13 * * *" | |
| concurrency: | |
| group: ${{ github.workflow }}-${{ github.event_name }}-${{ github.ref }} | |
| cancel-in-progress: true | |
| defaults: | |
| run: | |
| shell: bash | |
| env: | |
| MLFLOW_RUN_SLOW_TESTS: "true" | |
| MLFLOW_HOME: ${{ github.workspace }} | |
| PIP_EXTRA_INDEX_URL: https://download.pytorch.org/whl/cpu | |
| PIP_CONSTRAINT: ${{ github.workspace }}/requirements/constraints.txt | |
| jobs: | |
| docker-build: | |
| runs-on: ubuntu-latest | |
| timeout-minutes: 120 | |
| permissions: | |
| contents: read | |
| if: (github.event_name == 'schedule' && github.repository == 'mlflow/dev') || github.event_name == 'workflow_dispatch' || (github.event_name == 'pull_request' && github.event.pull_request.draft == false) | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| group: [1, 2, 3] | |
| include: | |
| - splits: 3 | |
| steps: | |
| - uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2 | |
| - uses: ./.github/actions/free-disk-space | |
| - uses: ./.github/actions/untracked | |
| - uses: ./.github/actions/setup-python | |
| - uses: ./.github/actions/setup-pyenv | |
| - uses: ./.github/actions/setup-java | |
| - name: Install dependencies | |
| run: | | |
| uv sync --extra extras | |
| uv pip install \ | |
| -r requirements/test-requirements.txt \ | |
| -r requirements/extra-ml-requirements.txt \ | |
| langchain_experimental tf-keras "pyarrow<18" | |
| - uses: ./.github/actions/show-versions | |
| - uses: ./.github/actions/pipdeptree | |
| - name: Run tests | |
| run: | | |
| uv run --no-sync pytest --splits=${{ matrix.splits }} --group=${{ matrix.group }} tests/pyfunc/docker |