Integ test updates

Author: samruds

src/sagemaker/serve/builder/transformers_builder.py

@@ -143,6 +143,11 @@ def _create_transformers_model(self) -> Type[Model]:
 
         logger.info("Detected %s. Proceeding with the the deployment.", self.image_uri)
 
+        if not pysdk_model:
+            raise ValueError(
+                "Ensure model has a framework - PyTorch or TensorFlow/Keras in it's metadata"
+            )
+
         if not pysdk_model.image_uri:
             pysdk_model.image_uri = self.image_uri
 

tests/integ/sagemaker/serve/test_serve_tei.py

@@ -88,7 +88,10 @@ def model_builder(request):
     return request.getfixturevalue(request.param)
 
 
-@pytest.mark.skipif(PYTHON_VERSION_IS_NOT_310)
+@pytest.mark.skipif(
+    PYTHON_VERSION_IS_NOT_310,
+    reason="Testing feature needs latest metadata",
+)
 @pytest.mark.parametrize("model_builder", ["model_builder_model_schema_builder"], indirect=True)
 def test_tei_sagemaker_endpoint(sagemaker_session, model_builder, model_input):
     logger.info("Running in SAGEMAKER_ENDPOINT mode...")