Skip to content

Commit

Permalink
python dev/update_ml_package_versions.py (mlflow#5413)
Browse files Browse the repository at this point in the history
Signed-off-by: Jenkins <jenkins@databricks.com>

Co-authored-by: Jenkins <jenkins@databricks.com>
  • Loading branch information
mlflow-automation and Jenkins committed Feb 25, 2022
1 parent e8b4aec commit 32bb686
Showing 1 changed file with 10 additions and 10 deletions.
20 changes: 10 additions & 10 deletions mlflow/ml-package-versions.yml
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ pytorch:
models:
minimum: "1.4.0"
maximum: "1.10.1"
maximum: "1.10.2"
requirements: ["torchvision", "scikit-learn", "transformers"]
run: |
pytest tests/pytorch/test_pytorch_model_export.py --large
Expand All @@ -41,7 +41,7 @@ pytorch-lightning:
autologging:
minimum: "1.0.5"
maximum: "1.5.9"
maximum: "1.5.10"
requirements: ["torchvision", "scikit-learn"]
run: |
pytest tests/pytorch/test_pytorch_autolog.py --large
Expand All @@ -54,7 +54,7 @@ tensorflow:
models:
minimum: "2.0.0"
maximum: "2.7.0"
maximum: "2.8.0"
requirements:
"< 2.2": ["h5py<3.0", "scikit-learn"]
# Requirements to run tests for keras
Expand All @@ -70,7 +70,7 @@ tensorflow:
autologging:
minimum: "2.0.0"
maximum: "2.7.0"
maximum: "2.8.0"
requirements:
"< 2.2": ["h5py<3.0"]
"== dev": ["scikit-learn"]
Expand All @@ -87,7 +87,7 @@ keras:

models:
minimum: "2.3.0"
maximum: "2.7.0"
maximum: "2.8.0"
requirements:
"<= 2.3.1": ["tensorflow==2.2.1", "scikit-learn", "pyspark", "pyarrow", "transformers<=4.11.3"]
"> 2.3.1, < 2.6.0": ["tensorflow<2.5.0", "scikit-learn", "pyspark", "pyarrow", "transformers"]
Expand All @@ -97,7 +97,7 @@ keras:
autologging:
minimum: "2.3.0"
maximum: "2.7.0"
maximum: "2.8.0"
requirements:
# keras 2.3.1 is incompatible with tensorflow > 2.2.1 and causes the issue reported here:
# https://github.com/tensorflow/tensorflow/issues/38589
Expand Down Expand Up @@ -229,7 +229,7 @@ onnx:
models:
minimum: "1.5.0"
maximum: "1.10.2"
maximum: "1.11.0"
requirements: ["onnxruntime", "torch", "scikit-learn"]
run: |
pytest tests/onnx/test_onnx_model_export.py --large
Expand All @@ -242,7 +242,7 @@ spacy:
models:
minimum: "2.2.4"
maximum: "3.2.1"
maximum: "3.2.2"
requirements: ["scikit-learn"]
run: |
pytest tests/spacy/test_spacy_model_export.py --large
Expand All @@ -255,13 +255,13 @@ statsmodels:
models:
minimum: "0.11.1"
maximum: "0.13.1"
maximum: "0.13.2"
run: |
pytest tests/statsmodels/test_statsmodels_model_export.py --large
autologging:
minimum: "0.11.1"
maximum: "0.13.1"
maximum: "0.13.2"
run: |
pytest tests/statsmodels/test_statsmodels_autolog.py --large
Expand Down

0 comments on commit 32bb686

Please sign in to comment.