Skip to content

Commit d5f9003

Browse files
author
Roja Reddy Sareddy
committed
Fix: fix sagemaker-serve tests
1 parent 1b27474 commit d5f9003

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

sagemaker-serve/tests/integ/test_model_customization_deployment.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -255,7 +255,7 @@ def test_sft_trainer_build(self, training_job_name):
255255
model="meta-textgeneration-llama-3-2-1b-instruct",
256256
training_dataset="s3://dummy/data.jsonl",
257257
accept_eula=True,
258-
model_package_group_name="test-group"
258+
model_package_group="test-group"
259259
)
260260
trainer._latest_training_job = training_job
261261

@@ -282,7 +282,7 @@ def test_dpo_trainer_build(self, training_job_name):
282282
model="meta-textgeneration-llama-3-2-1b-instruct",
283283
training_dataset="s3://dummy/data.jsonl",
284284
accept_eula=True,
285-
model_package_group_name="test-group"
285+
model_package_group="test-group"
286286
)
287287
trainer._latest_training_job = training_job
288288

0 commit comments

Comments
 (0)