fix: lock SKLearn framework version to latest for integ test and add assertion for ValueError in numpy deserializer unit test

Author: benieric

tests/integ/sagemaker/experiments/test_run.py

@@ -693,7 +693,7 @@ def _generate_estimator(
         sagemaker_client_config=sagemaker_client_config,
     )
     return SKLearn(
-        framework_version="0.23-1",
+        framework_version="1.2-1",
         entry_point=_ENTRY_POINT_PATH,
         dependencies=[sdk_tar],
         role=execution_role,

tests/unit/sagemaker/deserializers/test_deserializers.py

@@ -119,8 +119,9 @@ def test_numpy_deserializer_from_json(numpy_deserializer):
 # Sadly, ragged arrays work fine in JSON (giving us a 1D array of Python lists)
 def test_numpy_deserializer_from_json_ragged(numpy_deserializer):
     stream = io.BytesIO(b"[[1,2,3],\n[4,5,6,7]]")
-    array = numpy_deserializer.deserialize(stream, "application/json")
-    assert np.array_equal(array, np.array([[1, 2, 3], [4, 5, 6, 7]]))
+    with pytest.raises(ValueError) as error:
+        numpy_deserializer.deserialize(stream, "application/json")
+    assert "requested array has an inhomogeneous shape" in str(error)
 
 
 def test_numpy_deserializer_from_json_alpha():