diff --git a/pyproject.toml b/pyproject.toml index 1c60e786e2..696af88598 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -38,7 +38,7 @@ dependencies = [ "google-pasta", "importlib-metadata>=1.4.0,<7.0", "jsonschema", - "numpy>=1.9.0,<2.0", + "numpy==2.0", "packaging>=20.0", "pandas", "pathos", diff --git a/requirements/extras/test_requirements.txt b/requirements/extras/test_requirements.txt index 1592576a47..e0e980ac36 100644 --- a/requirements/extras/test_requirements.txt +++ b/requirements/extras/test_requirements.txt @@ -1,5 +1,5 @@ tox==3.24.5 -numpy>=1.24.0 +numpy==2.0 build[virtualenv]==1.2.1 flake8==4.0.1 pytest==6.2.5 diff --git a/src/sagemaker/serve/utils/conda_in_process.yml b/src/sagemaker/serve/utils/conda_in_process.yml index 7336130118..957dcf3926 100644 --- a/src/sagemaker/serve/utils/conda_in_process.yml +++ b/src/sagemaker/serve/utils/conda_in_process.yml @@ -12,7 +12,7 @@ dependencies: - boto3>=1.34.142,<2.0 - cloudpickle==2.2.1 - google-pasta - - numpy>=1.9.0,<2.0 + - numpy==2.0 - protobuf>=3.12,<5.0 - smdebug_rulesconfig==1.0.1 - importlib-metadata>=1.4.0,<7.0 diff --git a/tests/data/serve_resources/mlflow/pytorch/conda.yaml b/tests/data/serve_resources/mlflow/pytorch/conda.yaml index be61456197..de0b863025 100644 --- a/tests/data/serve_resources/mlflow/pytorch/conda.yaml +++ b/tests/data/serve_resources/mlflow/pytorch/conda.yaml @@ -11,7 +11,7 @@ dependencies: - defusedxml==0.7.1 - dill==0.3.8 - gmpy2==2.1.2 - - numpy==1.26.4 + - numpy==2.0 - opt-einsum==3.3.0 - packaging==24.0 - pandas==2.2.1 diff --git a/tests/data/serve_resources/mlflow/pytorch/requirements.txt b/tests/data/serve_resources/mlflow/pytorch/requirements.txt index 0446ed5053..76c1b05287 100644 --- a/tests/data/serve_resources/mlflow/pytorch/requirements.txt +++ b/tests/data/serve_resources/mlflow/pytorch/requirements.txt @@ -5,7 +5,7 @@ cloudpickle==2.2.1 defusedxml==0.7.1 dill==0.3.8 gmpy2==2.1.2 -numpy==1.24.4 +numpy==2.0 opt-einsum==3.3.0 packaging==21.3 pandas==2.2.1 diff --git a/tests/data/serve_resources/mlflow/tensorflow/conda.yaml b/tests/data/serve_resources/mlflow/tensorflow/conda.yaml index 90d8c300a0..4e8e060321 100644 --- a/tests/data/serve_resources/mlflow/tensorflow/conda.yaml +++ b/tests/data/serve_resources/mlflow/tensorflow/conda.yaml @@ -6,6 +6,6 @@ dependencies: - pip: - mlflow==2.11.1 - cloudpickle==2.2.1 - - numpy==1.26.4 + - numpy==2.0 - tensorflow==2.16.1 name: mlflow-env diff --git a/tests/data/serve_resources/mlflow/tensorflow/requirements.txt b/tests/data/serve_resources/mlflow/tensorflow/requirements.txt index ff99d3b92e..d633a8e813 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 -numpy==1.26.4 +numpy==2.0 tensorflow==2.16.1 diff --git a/tests/data/serve_resources/mlflow/xgboost/conda.yaml b/tests/data/serve_resources/mlflow/xgboost/conda.yaml index 44ca3c4c2e..92ed4106f5 100644 --- a/tests/data/serve_resources/mlflow/xgboost/conda.yaml +++ b/tests/data/serve_resources/mlflow/xgboost/conda.yaml @@ -6,7 +6,7 @@ dependencies: - pip: - mlflow==2.11.1 - lz4==4.3.2 - - numpy==1.26.4 + - numpy==2.0 - pandas==2.2.1 - psutil==5.9.8 - scikit-learn==1.3.2 diff --git a/tests/data/serve_resources/mlflow/xgboost/requirements.txt b/tests/data/serve_resources/mlflow/xgboost/requirements.txt index 1130dcaec5..3c0f14729e 100644 --- a/tests/data/serve_resources/mlflow/xgboost/requirements.txt +++ b/tests/data/serve_resources/mlflow/xgboost/requirements.txt @@ -1,6 +1,6 @@ mlflow==2.13.2 lz4==4.3.2 -numpy==1.24.4 +numpy==2.0 pandas==2.0.3 psutil==5.9.8 scikit-learn==1.3.2 diff --git a/tests/unit/sagemaker/jumpstart/constants.py b/tests/unit/sagemaker/jumpstart/constants.py index d22428f4f0..0d0f7e65b0 100644 --- a/tests/unit/sagemaker/jumpstart/constants.py +++ b/tests/unit/sagemaker/jumpstart/constants.py @@ -7467,7 +7467,7 @@ "inference_dependencies": [], "inference_vulnerabilities": [], "training_vulnerable": False, - "training_dependencies": ["numpy==1.19.5", "opencv_python==4.0.1.23"], + "training_dependencies": ["numpy==2.0", "opencv_python==4.0.1.23"], "training_vulnerabilities": [], "deprecated": False, "hyperparameters": [ diff --git a/tests/unit/sagemaker/serve/detector/test_dependency_manager.py b/tests/unit/sagemaker/serve/detector/test_dependency_manager.py index 491968dd25..f134c43d8e 100644 --- a/tests/unit/sagemaker/serve/detector/test_dependency_manager.py +++ b/tests/unit/sagemaker/serve/detector/test_dependency_manager.py @@ -21,7 +21,7 @@ DEPENDENCY_LIST = [ "requests==2.26.0", - "numpy>=1.20.0", + "numpy=2.0", "pandas<=1.3.3", "matplotlib<3.5.0", "scikit-learn>0.24.1",