chore: import upstream snapshot with attribution
This commit is contained in:
Executable
+51
@@ -0,0 +1,51 @@
|
||||
#!/usr/bin/env bash
|
||||
|
||||
# Executes a subset of mlflow tests that is supported with fewer dependencies than the core mlflow package.
|
||||
# Tests include most client interactions and compatibility points with the mlflow plugins around tracking, projects, models, deployments, and the cli.
|
||||
|
||||
# The SQL alchemy store's dependencies are added for a base client/store that can be tested against.
|
||||
# A different example client/store with a minimal dependency footprint could also work for this purpose.
|
||||
|
||||
set -x
|
||||
# Set err=1 if any commands exit with non-zero status as described in
|
||||
# https://stackoverflow.com/a/42219754
|
||||
err=0
|
||||
trap 'err=1' ERR
|
||||
export MLFLOW_SKINNY='true'
|
||||
|
||||
pytest tests/test_skinny_client_omits_sql_libs.py
|
||||
|
||||
# After verifying skinny client does not include store specific requirements,
|
||||
# we are installing sqlalchemy store requirements as our example store for the test suite.
|
||||
# SQL Alchemy serves as a simple, fully featured option to test skinny client store scenarios.
|
||||
uv pip install --system sqlalchemy alembic cryptography
|
||||
|
||||
# Given the example store does not delete dependencies, we verify non store related dependencies
|
||||
# after the example store setup. This verifies both the example store and skinny client do not add
|
||||
# unintended libraries.
|
||||
pytest tests/test_skinny_client_omits_data_science_libs.py
|
||||
|
||||
# Verify that mlflow.types.chat can be imported without numpy (issue #21779)
|
||||
pytest tests/test_skinny_client_anthropic_import.py
|
||||
|
||||
# Install numpy that is required by mlflow.types.schema and pre-installed in DBR.
|
||||
uv pip install --system numpy
|
||||
|
||||
pytest \
|
||||
tests/test_runs.py \
|
||||
tests/tracking/test_client.py \
|
||||
tests/tracking/test_tracking.py \
|
||||
tests/projects/test_projects.py \
|
||||
tests/deployments/test_cli.py \
|
||||
tests/deployments/test_deployments.py \
|
||||
tests/projects/test_projects_cli.py \
|
||||
tests/utils/test_requirements_utils.py::test_infer_requirements_excludes_mlflow \
|
||||
tests/utils/test_search_utils.py \
|
||||
tests/store/tracking/test_file_store.py \
|
||||
tests/utils/test_doctor.py \
|
||||
--import-mode=importlib
|
||||
|
||||
uv pip install --system pandas
|
||||
pytest tests/test_skinny_client_autolog_without_scipy.py
|
||||
|
||||
test $err = 0
|
||||
Reference in New Issue
Block a user