Skip to content

Commit 5b9f374

Browse files
author
Xiong Zeng
committed
Adjust formatting
1 parent c06ef53 commit 5b9f374

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

src/sagemaker/serve/builder/model_builder.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -244,7 +244,8 @@ class ModelBuilder(Triton, DJL, JumpStart, TGI, Transformers):
244244
default=None, metadata={"help": "Define the model server to deploy to."}
245245
)
246246
model_metadata: Optional[Dict[str, Any]] = field(
247-
default=None, metadata={"help": "Define the model metadata to override, currently supports `HF_TASK`"}
247+
default=None,
248+
metadata={"help": "Define the model metadata to override, currently supports `HF_TASK`"},
248249
)
249250

250251
def _build_validations(self):

tests/unit/sagemaker/serve/builder/test_model_builder.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1715,4 +1715,4 @@ def test_build_task_override_with_invalid_model_provided(
17151715
model=invalid_model_id, model_metadata={"HF_TASK": provided_task}
17161716
)
17171717
with self.assertRaises(Exception):
1718-
model_builder.build(sagemaker_session=mock_session)
1718+
model_builder.build(sagemaker_session=mock_session)

0 commit comments

Comments
 (0)