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",