Cloudpickle update

nargokul · nargokul · commit dad25af83a39 · 2024-12-10T18:33:10.000-08:00
diff --git a/requirements/extras/test_requirements.txt b/requirements/extras/test_requirements.txt
@@ -24,7 +24,7 @@ Jinja2==3.1.4
 pyvis==0.2.1
 pandas==1.4.4
 scikit-learn==1.3.0
-cloudpickle>=2.2.1
+cloudpickle>=2.2.1,<=3.0.0
 jsonpickle<4.0.0
 PyYAML==6.0
 # TODO find workaround
diff --git a/tests/data/serve_resources/mlflow/pytorch/requirements.txt b/tests/data/serve_resources/mlflow/pytorch/requirements.txt
@@ -1,7 +1,7 @@
 mlflow==2.13.2
 astunparse==1.6.3
 cffi==1.16.0
-cloudpickle>=2.2.1
+cloudpickle>=2.2.1,<=3.0.0
 defusedxml==0.7.1
 dill==0.3.8
 gmpy2==2.1.2
diff --git a/tests/data/serve_resources/mlflow/tensorflow/requirements.txt b/tests/data/serve_resources/mlflow/tensorflow/requirements.txt
@@ -1,4 +1,4 @@
 mlflow==2.13.2
-cloudpickle>=2.2.1
+cloudpickle>=2.2.1,<=3.0.0
 numpy==1.26.4
 tensorflow==2.16.1
