update SKLearn image URI config

Roja Reddy Sareddy · Roja Reddy Sareddy · commit 7acb57ee7e05 · 2025-10-07T13:02:29.000-07:00
diff --git a/tests/integ/sagemaker/serve/test_serve_mlflow_tensorflow_flavor_happy.py b/tests/integ/sagemaker/serve/test_serve_mlflow_tensorflow_flavor_happy.py
@@ -106,7 +106,7 @@ def tensorflow_schema_builder(custom_request_translator, custom_response_transla
 @pytest.mark.skipif(
     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",
+    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",
 )
 def test_happy_tensorflow_sagemaker_endpoint_with_tensorflow_serving(
     sagemaker_session,

Original file line number	Diff line number	Diff line change
`@@ -106,7 +106,7 @@ def tensorflow_schema_builder(custom_request_translator, custom_response_transla`
`106`	`106`	`@pytest.mark.skipif(`
`107`	`107`	`PYTHON_VERSION_IS_NOT_310,`
`108`	`108`	`np.__version__ >= "2.0.0",`
`109`		`- reason="The goal of these test are to test the serving components of our feature",`
	`109`	`+ 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",`
`110`	`110`	`)`
`111`	`111`	`def test_happy_tensorflow_sagemaker_endpoint_with_tensorflow_serving(`
`112`	`112`	`sagemaker_session,`