fix: pin scikit-learn in mlflow UAT (#37)

canonical · Oct 2, 2023 · 6a75524 · 6a75524
1 parent f2cc4dc
commit 6a75524
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/tests/notebooks/mlflow/requirements.in b/tests/notebooks/mlflow/requirements.in
@@ -3,4 +3,6 @@ boto3
 minio<7.0
 # pin the client to match the version of the deployed MLflow server
 mlflow==2.1.1
+# pin to ensure compatibility with the installed mlflow client
+scikit-learn<1.2
 tenacity
diff --git a/tests/notebooks/mlflow/requirements.txt b/tests/notebooks/mlflow/requirements.txt
@@ -154,8 +154,9 @@ requests==2.31.0
     #   mlflow
 s3transfer==0.6.2
     # via boto3
-scikit-learn==1.3.0
+scikit-learn==1.1.3
     # via
+    #   -r requirements.in
     #   mlflow
     #   shap
 scipy==1.10.1