diff --git a/pyproject.toml b/pyproject.toml index 9ad30d20a1..242b66f73e 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -33,13 +33,13 @@ classifiers = [ dependencies = [ "attrs>=23.1.0,<24", "boto3>=1.35.75,<2.0", - "cloudpickle>=2.2.1", + "cloudpickle==2.2.1", "docker", "fastapi", "google-pasta", "importlib-metadata>=1.4.0,<7.0", "jsonschema", - "numpy>=1.9.0,<3.0", + "numpy>=1.9.0,<2.0", "omegaconf>=2.2,<2.3", "packaging>=20.0", "pandas", diff --git a/requirements/extras/test_requirements.txt b/requirements/extras/test_requirements.txt index 9664a63e1d..fe31300c22 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 jsonpickle<4.0.0 PyYAML==6.0 # TODO find workaround diff --git a/src/sagemaker/serve/utils/conda_in_process.yml b/src/sagemaker/serve/utils/conda_in_process.yml index 7336130118..61badaa52f 100644 --- a/src/sagemaker/serve/utils/conda_in_process.yml +++ b/src/sagemaker/serve/utils/conda_in_process.yml @@ -39,7 +39,7 @@ dependencies: - certifi==2022.12.7 - charset-normalizer>=3.1.0 - click>=8.1.3 - - cloudpickle>=2.2.1 + - cloudpickle==2.2.1 - colorama>=0.4.4 - contextlib2>=21.6.0 - decorator>=5.1.1 diff --git a/src/sagemaker/serve/utils/in_process_requirements.txt b/src/sagemaker/serve/utils/in_process_requirements.txt index 5a620937eb..e356e1720d 100644 --- a/src/sagemaker/serve/utils/in_process_requirements.txt +++ b/src/sagemaker/serve/utils/in_process_requirements.txt @@ -7,7 +7,7 @@ cachetools>=5.3.0 certifi==2024.7.4 charset-normalizer>=3.1.0 click>=8.1.3 -cloudpickle>=2.2.1 +cloudpickle==2.2.1 colorama>=0.4.4 contextlib2>=21.6.0 decorator>=5.1.1 diff --git a/tests/data/serve_resources/mlflow/pytorch/requirements.txt b/tests/data/serve_resources/mlflow/pytorch/requirements.txt index 12bae64c99..0446ed5053 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 cd1cbee65b..ff99d3b92e 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