update cloudpickle version to >=2.2.1

Author: liujiaorr

pyproject.toml

@@ -33,7 +33,7 @@ classifiers = [
 dependencies = [
   "attrs>=23.1.0,<24",
   "boto3>=1.34.142,<2.0",
-  "cloudpickle==2.2.1",
+  "cloudpickle>=2.2.1",
   "docker",
   "google-pasta",
   "importlib-metadata>=1.4.0,<7.0",

requirements/extras/test_requirements.txt

@@ -24,7 +24,7 @@ Jinja2==3.1.4
 pyvis==0.2.1
 pandas>=1.3.5,<1.5
 scikit-learn==1.3.0
-cloudpickle==2.2.1
+cloudpickle>=2.2.1
 PyYAML==6.0
 # TODO find workaround
 xgboost>=1.6.2,<=1.7.6

src/sagemaker/serve/utils/conda_in_process.yml

@@ -10,7 +10,7 @@ dependencies:
     - pip>=23.0.1
     - attrs>=23.1.0,<24
     - boto3>=1.34.142,<2.0
-    - cloudpickle==2.2.1
+    - cloudpickle>=2.2.1
     - google-pasta
     - numpy>=1.9.0,<2.0
     - protobuf>=3.12,<5.0

tests/data/serve_resources/mlflow/pytorch/conda.yaml

@@ -7,7 +7,7 @@ dependencies:
   - mlflow==2.10.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/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/conda.yaml

@@ -5,7 +5,7 @@ dependencies:
 - pip<=23.3.1
 - pip:
   - mlflow==2.11.1
-  - cloudpickle==2.2.1
+  - cloudpickle>=2.2.1
   - numpy==1.26.4
   - tensorflow==2.16.1
 name: mlflow-env

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