Skip to content

Commit 7f36702

Browse files
committed
Fix formatting
1 parent 48205ad commit 7f36702

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/sagemaker/serve/builder/tei_builder.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ def _set_to_tgi(self):
8585
logger.warning(messaging)
8686
self.model_server = ModelServer.TGI
8787

88-
def _create_tei_model(self, **kwargs) -> Type[Model]:
88+
def _create_tei_model(self, **kwargs) -> Type[Model]:
8989
"""Placeholder docstring"""
9090
if self.nb_instance_type and "instance_type" not in kwargs:
9191
kwargs.update({"instance_type": self.nb_instance_type})
@@ -94,7 +94,7 @@ def _create_tei_model(self, **kwargs) -> Type[Model]:
9494
self.image_uri = image_uris.retrieve(
9595
"huggingface-tei",
9696
image_scope="inference",
97-
instance_type=kwargs.get("instance_type")
97+
instance_type=kwargs.get("instance_type"),
9898
)
9999

100100
pysdk_model = HuggingFaceModel(

0 commit comments

Comments
 (0)