ModelBuilder-JS

Author: Jonathan Makunga

src/sagemaker/serve/builder/jumpstart_builder.py

@@ -439,6 +439,9 @@ def set_deployment_config(self, config_name: Optional[str]) -> None:
                 The name of the deployment config. Set to None to unset
                 any existing config that is applied to the model.
         """
+        if self.pysdk_model is None:
+            self.pysdk_model = self._create_pre_trained_js_model()
+
         self.pysdk_model.set_deployment_config(config_name)
 
     def get_deployment_config(self) -> Optional[Dict[str, Any]]:
@@ -447,10 +450,16 @@ def get_deployment_config(self) -> Optional[Dict[str, Any]]:
         Returns:
             Union[Dict[str, Any], None]: Deployment config to apply to this model.
         """
+        if self.pysdk_model is None:
+            self.pysdk_model = self._create_pre_trained_js_model()
+
         return self.pysdk_model.deployment_config
 
     def display_benchmark_metrics(self):
         """Display Markdown Benchmark Metrics for deployment configs."""
+        if self.pysdk_model is None:
+            self.pysdk_model = self._create_pre_trained_js_model()
+
         self.pysdk_model.display_benchmark_metrics()
 
     def list_deployment_configs(self) -> List[Dict[str, Any]]:
@@ -459,6 +468,9 @@ def list_deployment_configs(self) -> List[Dict[str, Any]]:
         Returns:
             List[Dict[str, Any]]: A list of deployment configs.
         """
+        if self.pysdk_model is None:
+            self.pysdk_model = self._create_pre_trained_js_model()
+
         return self.pysdk_model.list_deployment_configs()
 
     def _build_for_jumpstart(self):
@@ -467,7 +479,11 @@ def _build_for_jumpstart(self):
         self.secret_key = None
         self.jumpstart = True
 
-        pysdk_model = self._create_pre_trained_js_model()
+        pysdk_model = (
+            self.pysdk_model
+            if self.pysdk_model is not None
+            else self._create_pre_trained_js_model()
+        )
 
         image_uri = pysdk_model.image_uri