Skip to content

Commit f6f8116

Browse files
committed
Fix formatting
1 parent 29ea1c5 commit f6f8116

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/sagemaker/serve/builder/tei_builder.py

Lines changed: 2 additions & 2 deletions
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})
@@ -95,7 +95,7 @@ def _create_tei_model(self, **kwargs) -> Type[Model]:
9595
"huggingface-tei",
9696
image_scope="inference",
9797
instance_type=kwargs.get("instance_type"),
98-
region=self.sagemaker_session.boto_region_name
98+
region=self.sagemaker_session.boto_region_name,
9999
)
100100

101101
pysdk_model = HuggingFaceModel(

0 commit comments

Comments
 (0)