Skip to content

Commit 98161ac

Browse files
rsareddy0329Roja Reddy Sareddy
andauthored
Fix: fix sagemaker-serve tests (#5407)
* fix: Fix the recipe selection for multiple recipe scenario * fix: Fix the recipe selection for multiple recipe scenario * fix: Hyperparameter issue fixes, validate s3 output path,additional unit tests * Fix: Add validation to bedrock reward models * Fix: Add validation to bedrock reward models * Fix: Add allow list for bedrock eval models * Fix: Add allow list for bedrock eval models * Fix: Bug fixes for s3 path validation, mlflow app creation * Fix: Update Legal verbiage, and allowed reward model ids based on region * Fix: Update model_package_group_name to model_package_group in all trianers to maintain consistency * Fix: fix sagemaker-serve tests --------- Co-authored-by: Roja Reddy Sareddy <rsareddy@amazon.com>
1 parent 1ab6d30 commit 98161ac

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)