Cloudpickle Revert

Author: nargokul

pyproject.toml

@@ -33,7 +33,7 @@ classifiers = [
 dependencies = [
   "attrs>=23.1.0,<24",
   "boto3>=1.35.75,<2.0",
-  "cloudpickle>=2.2.1",
+  "cloudpickle==2.2.1",
   "docker",
   "fastapi",
   "google-pasta",

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
 jsonpickle<4.0.0
 PyYAML==6.0
 # TODO find workaround

src/sagemaker/serve/utils/conda_in_process.yml

@@ -39,7 +39,7 @@ dependencies:
             - certifi==2022.12.7
             - charset-normalizer>=3.1.0
             - click>=8.1.3
-            - cloudpickle>=2.2.1
+            - cloudpickle==2.2.1
             - colorama>=0.4.4
             - contextlib2>=21.6.0
             - decorator>=5.1.1

src/sagemaker/serve/utils/in_process_requirements.txt

@@ -7,7 +7,7 @@ cachetools>=5.3.0
 certifi==2024.7.4
 charset-normalizer>=3.1.0
 click>=8.1.3
-cloudpickle>=2.2.1
+cloudpickle==2.2.1
 colorama>=0.4.4
 contextlib2>=21.6.0
 decorator>=5.1.1

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
 defusedxml==0.7.1
 dill==0.3.8
 gmpy2==2.1.2

tests/data/serve_resources/mlflow/tensorflow/requirements.txt

@@ -1,4 +1,4 @@
 mlflow==2.13.2
-cloudpickle>=2.2.1
+cloudpickle==2.2.1
 numpy==1.26.4
 tensorflow==2.16.1