codestyle

Author: Bryannah Hernandez

src/sagemaker/serve/builder/djl_builder.py

@@ -472,10 +472,10 @@ def _auto_detect_container(self):
         if self.model:
             logger.info(
                 "Auto detect container url for the provided model and on instance %s",
-                self.instance_type,
+                self.nb_instance_type,
             )
             self.image_uri = auto_detect_container(
-                self.model, self.sagemaker_session.boto_region_name, self.instance_type
+                self.model, self.sagemaker_session.boto_region_name, self.nb_instance_type
             )
 
         elif self.inference_spec:
@@ -488,7 +488,7 @@ def _auto_detect_container(self):
             self.image_uri = auto_detect_container(
                 self.inference_spec.load(self.model_path),
                 self.sagemaker_session.boto_region_name,
-                self.instance_type,
+                self.nb_instance_type,
             )
         else:
             raise ValueError(

src/sagemaker/serve/mode/sagemaker_endpoint_mode.py

@@ -95,6 +95,7 @@ def prepare(
             upload_artifacts = self._upload_djl_artifacts(
                 model_path=model_path,
                 sagemaker_session=sagemaker_session,
+                secret_key=secret_key,
                 s3_model_data_url=s3_model_data_url,
                 image=image,
                 should_upload_artifacts=True,