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