Revert the change "Add Numpy 2.0 support" (#5307)

* Add numpy 2.0 support

* Add numpy 2.0 support

* Add numpy 2.0 support

* Add numpy 2.0 support

* Add numpy 2.0 support

* Fix incompatible_dependecies test

* Fix incompatible_dependecies test

* Fix incompatible_dependecies test

* Fix incompatible_dependecies test

* Fix incompatible_dependecies test

* update tensorflow artifacts

* update tensorflow artifacts

* update tensorflow artifacts

* testfile codestyle fixes

* testfile codestyle fixes

* update SKLearn image URI config

* update SKLearn image URI config

* docstyle fixes

* docstyle fixes

* numpy fixes

* numpy fixes

* numpy fixes

* numpy fixes

* numpy fixes

* numpy fixes

* numpy fixes

* numpy fixes

* numpy fix for slow test

* numpy fix for slow test

* numpy fix for slow test

* numpy fix for slow test

* Revert 'Add numpy 2.0 support'

* Revert 'Add numpy 2.0 support'

* Revert 'Add numpy 2.0 support'

---------

Co-authored-by: Roja Reddy Sareddy <[email protected]>
Co-authored-by: parknate@ <[email protected]>
Co-authored-by: Gokul Anantha Narayanan <[email protected]>

Author: 4 people

pyproject.toml

@@ -39,7 +39,7 @@ dependencies = [
   "google-pasta",
   "importlib-metadata>=1.4.0,<7.0",
   "jsonschema",
-  "numpy>=1.26.4,<2.3.3",
+  "numpy==1.26.4",
   "omegaconf>=2.2,<3",
   "packaging>=23.0,<25",
   "pandas",

requirements/extras/scipy_requirements.txt

@@ -1 +1 @@
-scipy==1.13.0
+scipy==1.11.3

requirements/extras/test_requirements.txt

@@ -1,5 +1,5 @@
 tox==3.24.5
-numpy>=2.0.0, <2.3.3
+numpy==1.26.4
 build[virtualenv]==1.2.1
 flake8==7.1.2
 pytest==6.2.5
@@ -23,8 +23,8 @@ requests==2.32.2
 sagemaker-experiments==0.1.35
 Jinja2==3.1.6
 pyvis==0.2.1
-pandas>=2.3.0
-scikit-learn==1.6.1
+pandas==1.4.4
+scikit-learn==1.3.0
 cloudpickle==2.2.1
 jsonpickle<4.0.0
 PyYAML>=6.0.1
@@ -44,7 +44,7 @@ onnx==1.17.0
 nbformat>=5.9,<6
 accelerate>=0.24.1,<=0.27.0
 schema==0.7.5
-tensorflow>=2.16.2,<=2.19.0
+tensorflow>=2.16.2,<=2.18.0
 mlflow>=2.14.2,<3
 huggingface_hub==0.26.2
 uvicorn>=0.30.1

src/sagemaker/image_uri_config/sklearn.json

@@ -388,54 +388,6 @@
                     "us-west-2": "246618743249"
                 },
                 "repository": "sagemaker-scikit-learn"
-            },
-            "1.4-2": {
-                "processors": [
-                    "cpu"
-                ],
-                "py_versions": [
-                    "py3"
-                ],
-                "registries": {
-                    "af-south-1": "510948584623",
-                    "ap-east-1": "651117190479",
-                    "ap-northeast-1": "354813040037",
-                    "ap-northeast-2": "366743142698",
-                    "ap-northeast-3": "867004704886",
-                    "ap-south-1": "720646828776",
-                    "ap-south-2": "628508329040",
-                    "ap-southeast-1": "121021644041",
-                    "ap-southeast-2": "783357654285",
-                    "ap-southeast-3": "951798379941",
-                    "ap-southeast-4": "106583098589",
-                    "ca-central-1": "341280168497",
-                    "ca-west-1": "190319476487",
-                    "cn-north-1": "450853457545",
-                    "cn-northwest-1": "451049120500",
-                    "eu-central-1": "492215442770",
-                    "eu-central-2": "680994064768",
-                    "eu-north-1": "662702820516",
-                    "eu-south-1": "978288397137",
-                    "eu-south-2": "104374241257",
-                    "eu-west-1": "141502667606",
-                    "eu-west-2": "764974769150",
-                    "eu-west-3": "659782779980",
-                    "il-central-1": "898809789911",
-                    "me-central-1": "272398656194",
-                    "me-south-1": "801668240914",
-                    "sa-east-1": "737474898029",
-                    "us-east-1": "683313688378",
-                    "us-east-2": "257758044811",
-                    "us-gov-east-1": "237065988967",
-                    "us-gov-west-1": "414596584902",
-                    "us-iso-east-1": "833128469047",
-                    "us-isob-east-1": "281123927165",
-                    "us-isof-east-1": "108575199400",
-                    "us-isof-south-1": "124985052026",
-                    "us-west-1": "746614075791",
-                    "us-west-2": "246618743249"
-                },
-                "repository": "sagemaker-scikit-learn"
             }
         }
     },

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>=2.0.0,<2.3.3
+    - numpy==1.26.4
     - protobuf>=3.12,<5.0
     - smdebug_rulesconfig==1.0.1
     - importlib-metadata>=1.4.0,<7.0
@@ -64,7 +64,7 @@ dependencies:
             - multiprocess>=0.70.14
             - networkx>=3.1
             - packaging>=23.1
-            - pandas>=2.3.0
+            - pandas>=1.5.3
             - pathos>=0.3.0
             - pillow>=9.5.0
             - platformdirs>=3.2.0

tests/data/remote_function/requirements.txt

@@ -1 +1 @@
-scipy==1.13.0
+scipy==1.11.3

tests/data/serve_resources/mlflow/pytorch/MLmodel

@@ -14,7 +14,7 @@ flavors:
     code: null
     model_data: data
     pytorch_version: 2.0.1+cu117
-mlflow_version: 2.10.2
+mlflow_version: 2.11.1
 model_size_bytes: 4971001
 model_uuid: 2d85043bbf504b1e9950e124c46a1719
 run_id: 98b8d2e2c0e74ab59f4c26f7cb3de233

tests/data/serve_resources/mlflow/pytorch/conda.yaml

@@ -2,23 +2,23 @@ channels:
 - conda-forge
 dependencies:
 - python=3.10.13
-- pip<=24.3
+- pip<=23.3.1
 - pip:
-  - mlflow>=2.16.1
+  - mlflow==2.10.2
   - astunparse==1.6.3
   - cffi==1.16.0
   - cloudpickle==2.2.1
   - defusedxml==0.7.1
   - dill==0.3.9
   - gmpy2==2.1.2
-  - numpy>=2.0.0,<2.3.3
+  - numpy==1.26.4
   - opt-einsum==3.3.0
   - packaging==24.0
-  - pandas>=2.3.0
+  - pandas==2.2.1
   - pyyaml==6.0.1
   - requests==2.31.0
   - torch>=2.6.0
   - torchvision>=0.17.0
   - tqdm==4.66.2
-  - scikit-learn==1.6.1
+  - scikit-learn==1.3.2
 name: mlflow-env

tests/data/serve_resources/mlflow/pytorch/requirements.txt

@@ -5,10 +5,10 @@ cloudpickle==2.2.1
 defusedxml==0.7.1
 dill==0.3.9
 gmpy2==2.1.2
-numpy>=2.0.0,<2.3.3
+numpy==1.26.4
 opt-einsum==3.3.0
 packaging>=23.0,<25
-pandas>=2.3.0
+pandas==2.2.1
 pyyaml==6.0.1
 requests==2.32.4
 torch>=2.6.0

tests/data/serve_resources/mlflow/tensorflow/conda.yaml

@@ -2,10 +2,10 @@ channels:
 - conda-forge
 dependencies:
 - python=3.10.13
-- pip<=24.3
+- pip<=23.3.1
 - pip:
-  - mlflow>=2.16.1
-  - cloudpickle>=2.2.1
-  - numpy>=1.26.4,<2.3.3
-  - tensorflow==2.18.0
+  - mlflow==2.11.1
+  - cloudpickle==2.2.1
+  - numpy==1.26.4
+  - tensorflow==2.16.1
 name: mlflow-env