Skip to content

Commit 4800eb1

Browse files
author
Bryannah Hernandez
committed
formatting changes
1 parent ca26cd2 commit 4800eb1

File tree

1 file changed

+6
-7
lines changed

1 file changed

+6
-7
lines changed

src/sagemaker/serve/builder/transformers_builder.py

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -353,7 +353,6 @@ def _build_for_transformers(self):
353353
if not os.path.exists(self.model_path):
354354
os.makedirs(self.model_path)
355355

356-
357356
code_path = Path(self.model_path).joinpath("code")
358357

359358
save_pkl(code_path, (self.inference_spec, self.schema_builder))
@@ -363,13 +362,13 @@ def _build_for_transformers(self):
363362

364363
self.secret_key = prepare_for_mms(
365364
model_path=self.model_path,
366-
shared_libs=self.shared_libs,
367-
dependencies=self.dependencies,
368-
session=self.sagemaker_session,
369-
image_uri=self.image_uri,
370-
inference_spec=self.inference_spec,
365+
shared_libs=self.shared_libs,
366+
dependencies=self.dependencies,
367+
session=self.sagemaker_session,
368+
image_uri=self.image_uri,
369+
inference_spec=self.inference_spec,
371370
)
372371

373-
self._build_transformers_env()
372+
self._build_transformers_env()
374373

375374
return self.pysdk_model

0 commit comments

Comments
 (0)