numpy fixes

Author: Roja Reddy Sareddy

tests/integ/sagemaker/serve/test_serve_mlflow_tensorflow_flavor_happy.py

@@ -107,15 +107,14 @@ def tensorflow_schema_builder(custom_request_translator, custom_response_transla
     PYTHON_VERSION_IS_NOT_310,
     np.__version__ >= "2.0.0",
     reason="The goal of these test are to test the serving components of our feature and \
-           the input model artifacts used here are generated with py310 and numpy<2",
+           the input model artifacts used in this specific test are generated with py310 and numpy<2.",
 )
 def test_happy_tensorflow_sagemaker_endpoint_with_tensorflow_serving(
     sagemaker_session,
     tensorflow_schema_builder,
     cpu_instance_type,
     test_data,
 ):
-    # TODO: Might need to add a new test for numpy 2.0
     logger.info("Running in SAGEMAKER_ENDPOINT mode...")
     caught_ex = None
 

tests/integ/sagemaker/serve/test_serve_mlflow_xgboost_flavor_happy.py

@@ -28,7 +28,7 @@
     XGBOOST_MLFLOW_RESOURCE_DIR,
     SERVE_SAGEMAKER_ENDPOINT_TIMEOUT,
     # SERVE_LOCAL_CONTAINER_TIMEOUT,
-    PYTHON_VERSION_IS_NOT_310,
+    # PYTHON_VERSION_IS_NOT_310,
 )
 from tests.integ.timeout import timeout
 from tests.integ.utils import cleanup_model_resources