From a7348c681bc97e10c14edf69196f06b8d010e8d4 Mon Sep 17 00:00:00 2001
From: Jeff Heckey <jheckey@gmail.com>
Date: Mon, 18 Nov 2024 12:55:34 -0800
Subject: [PATCH] fix: Pin cloudpickle==2.2.1 (#1044)

---
 setup.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 18bc591e1..2e1afe1c7 100644
--- a/setup.py
+++ b/setup.py
@@ -33,7 +33,9 @@
         "backoff",
         "boltons",
         "boto3>=1.28.53",
-        "cloudpickle>=3",
+        # SageMaker pinned cloudpickle==2.2.1
+        # see https://github.com/aws/sagemaker-python-sdk/issues/4871
+        "cloudpickle==2.2.1",
         "nest-asyncio",
         "networkx",
         "numpy",