Fix formatting

aws · May 17, 2024 · 7f36702 · 7f36702
1 parent 48205ad
commit 7f36702
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/sagemaker/serve/builder/tei_builder.py b/src/sagemaker/serve/builder/tei_builder.py
@@ -85,7 +85,7 @@ def _set_to_tgi(self):
             logger.warning(messaging)
             self.model_server = ModelServer.TGI
 
-    def _create_tei_model(self,  **kwargs) -> Type[Model]:
+    def _create_tei_model(self, **kwargs) -> Type[Model]:
         """Placeholder docstring"""
         if self.nb_instance_type and "instance_type" not in kwargs:
             kwargs.update({"instance_type": self.nb_instance_type})
@@ -94,7 +94,7 @@ def _create_tei_model(self,  **kwargs) -> Type[Model]:
             self.image_uri = image_uris.retrieve(
                 "huggingface-tei",
                 image_scope="inference",
-                instance_type=kwargs.get("instance_type")
+                instance_type=kwargs.get("instance_type"),
             )
 
         pysdk_model = HuggingFaceModel(