re-format lmi config switch

aws · Nov 22, 2024 · 33dcf96 · 33dcf96
1 parent 8b96155
commit 33dcf96
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/sagemaker/serve/builder/model_builder.py b/src/sagemaker/serve/builder/model_builder.py
@@ -1303,7 +1303,9 @@ def _model_builder_optimize_wrapper(
         if self._is_jumpstart_model_id():
             self.build(mode=self.mode, sagemaker_session=self.sagemaker_session)
             if self.pysdk_model:
-                self.pysdk_model.set_deployment_config(instance_type=instance_type, config_name="lmi")
+                self.pysdk_model.set_deployment_config(
+                    instance_type=instance_type, config_name="lmi"
+                )
             input_args = self._optimize_for_jumpstart(
                 output_path=output_path,
                 instance_type=instance_type,