Skip to content

Commit 90ae56c

Browse files
committed
Fix formatting
1 parent df63748 commit 90ae56c

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

tests/integ/sagemaker/serve/test_schema_builder.py

+3-5
Original file line numberDiff line numberDiff line change
@@ -152,7 +152,7 @@ def test_model_builder_happy_path_with_task_provided_local_schema_mode(
152152
"question-answering",
153153
"ml.m5.xlarge",
154154
),
155-
("deepset/roberta-base-squad2", "question-answering", "ml.m5.xlarge")
155+
("deepset/roberta-base-squad2", "question-answering", "ml.m5.xlarge"),
156156
],
157157
)
158158
def test_model_builder_happy_path_with_task_provided_remote_schema_mode(
@@ -208,12 +208,10 @@ def test_model_builder_happy_path_with_task_provided_remote_schema_mode(
208208
)
209209
@pytest.mark.parametrize(
210210
"model_id, task_provided, instance_type_provided",
211-
[
212-
("openai/whisper-large-v3", "automatic-speech-recognition", "ml.m5.xlarge")
213-
],
211+
[("openai/whisper-large-v3", "automatic-speech-recognition", "ml.m5.xlarge")],
214212
)
215213
def test_model_builder_happy_path_with_task_provided_remote_schema_mode_asr(
216-
model_id, task_provided, sagemaker_session, instance_type_provided
214+
model_id, task_provided, sagemaker_session, instance_type_provided
217215
):
218216
model_builder = ModelBuilder(
219217
model=model_id,

0 commit comments

Comments
 (0)