From 6d7d0663ef4e155052d76a3f2a216f23619c3e9d Mon Sep 17 00:00:00 2001
From: Gokul A <166456257+nargokul@users.noreply.github.com>
Date: Mon, 9 Dec 2024 14:50:20 -0800
Subject: [PATCH 1/3] Cloudpickle update

---
 tests/data/serve_resources/mlflow/pytorch/requirements.txt    | 2 +-
 tests/data/serve_resources/mlflow/tensorflow/requirements.txt | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/data/serve_resources/mlflow/pytorch/requirements.txt b/tests/data/serve_resources/mlflow/pytorch/requirements.txt
index 0446ed5053..12bae64c99 100644
--- 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
 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
index ff99d3b92e..cd1cbee65b 100644
--- 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
 numpy==1.26.4
 tensorflow==2.16.1

From d1b507266bf7833132acd741a29b99216fc692ae Mon Sep 17 00:00:00 2001
From: Gokul A <166456257+nargokul@users.noreply.github.com>
Date: Tue, 10 Dec 2024 18:29:51 -0800
Subject: [PATCH 2/3] Cloudpickle update

---
 pyproject.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 9ad30d20a1..45788d3cc8 100644
--- a/pyproject.toml
+++ b/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,<=3.0.0",
   "docker",
   "fastapi",
   "google-pasta",

From dad25af83a39e6c5a843fe259c4baeec6cecc511 Mon Sep 17 00:00:00 2001
From: Gokul A <166456257+nargokul@users.noreply.github.com>
Date: Tue, 10 Dec 2024 18:33:10 -0800
Subject: [PATCH 3/3] Cloudpickle update

---
 requirements/extras/test_requirements.txt                     | 2 +-
 tests/data/serve_resources/mlflow/pytorch/requirements.txt    | 2 +-
 tests/data/serve_resources/mlflow/tensorflow/requirements.txt | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/requirements/extras/test_requirements.txt b/requirements/extras/test_requirements.txt
index 9664a63e1d..3ae90e0151 100644
--- 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
index 12bae64c99..418e28fd4a 100644
--- 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
index cd1cbee65b..bf2269bd0e 100644
--- 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