Debugging issue

Author: samruds

src/sagemaker/serve/builder/model_builder.py

@@ -451,6 +451,7 @@ def _create_model(self):
         self._original_register = self.pysdk_model.register
         self.pysdk_model.register = self._model_builder_register_wrapper
         self.model_package = None
+        logger.info("Model %s registered", self.pysdk_model)
         return self.pysdk_model
 
     @_capture_telemetry("register")
@@ -565,6 +566,7 @@ def _build_for_torchserve(self) -> Type[Model]:
         )
 
         self._prepare_for_mode()
+        logger.info("Building for torchserve")
 
         return self._create_model()
 

tests/integ/sagemaker/serve/test_serve_pt_happy.py

@@ -198,20 +198,19 @@ def test_happy_pytorch_sagemaker_endpoint(
     logger.info("Running in SAGEMAKER_ENDPOINT mode...")
     caught_ex = None
 
-    iam_client = sagemaker_session.boto_session.client("iam")
-    role_arn = iam_client.get_role(RoleName=ROLE_NAME)["Role"]["Arn"]
+    # iam_client = sagemaker_session.boto_session.client("iam")
+    # role_arn = iam_client.get_role(RoleName=ROLE_NAME)["Role"]["Arn"]
 
-    model = model_builder.build(
-        mode=Mode.SAGEMAKER_ENDPOINT, role_arn=role_arn, sagemaker_session=sagemaker_session
-    )
+    model = model_builder.build(mode=Mode.LOCAL_CONTAINER, sagemaker_session=sagemaker_session)
 
     with timeout(minutes=SERVE_SAGEMAKER_ENDPOINT_TIMEOUT):
         try:
             logger.info("Deploying and predicting in SAGEMAKER_ENDPOINT mode...")
-            predictor = model.deploy(instance_type=cpu_instance_type, initial_instance_count=1)
+            predictor = model.deploy()
             logger.info("Endpoint successfully deployed.")
             predictor.predict(test_image)
         except Exception as e:
+            logger.exception("test failed")
             caught_ex = e
         finally:
             cleanup_model_resources(