Add Numpy 2.0 support

For commit: aws/sagemaker-python-sdk-staging@99210b2

Tested by running sagemaker-serve unit tests

Author: zhaoqizqwang

requirements/extras/test_requirements.txt

@@ -4,8 +4,11 @@ pytest-xdist
 mock
 pydantic==2.11.9
 pydantic_core==2.33.2
-pandas
+pandas>=2.3.0
+numpy>=2.0.0, <3.0
+scikit-learn==1.6.1
 scipy
 omegaconf
 graphene
-typing_extensions>=4.9.0
+typing_extensions>=4.9.0
+tensorflow>=2.16.2,<=2.19.0

sagemaker-serve/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.0,<3.0
     - 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>=1.5.3
+            - pandas>=2.3.0
             - pathos>=0.3.0
             - pillow>=9.5.0
             - platformdirs>=3.2.0