fix: Fix failing train tests for v3 (aws#5814)

mohamedzeidan2021 · Mohamed Zeidan · web-flow · commit 1885e4c513a9 · 2026-05-01T10:37:21.000-07:00
* fix: Fix failing train tests for v3
---------

Co-authored-by: Mohamed Zeidan &lt;zeidmo@amazon.com&gt;
diff --git a/sagemaker-train/tests/integ/jumpstart/test_jumpstart_train.py b/sagemaker-train/tests/integ/jumpstart/test_jumpstart_train.py
@@ -30,7 +30,7 @@
             },
             # Override default instance type; the model's default
             # (ml.p3.2xlarge) is deprecated.
-            "compute": Compute(instance_type="ml.g5.xlarge"),
+            "compute": Compute(instance_type="ml.g4dn.xlarge"),
         },
         {"model_id": "xgboost-classification-model"},
         {"model_id": "catboost-regression-model"},
diff --git a/sagemaker-train/tests/integ/train/test_llm_as_judge_base_model_fix.py b/sagemaker-train/tests/integ/train/test_llm_as_judge_base_model_fix.py
@@ -76,6 +76,7 @@
 }
 
 
+@pytest.mark.serial
 class TestLLMAsJudgeBaseModelFix:
     """Integration test for base model fix in LLMAsJudgeEvaluator"""
 

Original file line number	Diff line number	Diff line change
`@@ -76,6 +76,7 @@`
`76`	`76`	`}`
`77`	`77`
`78`	`78`
	`79`	`+@pytest.mark.serial`
`79`	`80`	`class TestLLMAsJudgeBaseModelFix:`
`80`	`81`	`"""Integration test for base model fix in LLMAsJudgeEvaluator"""`
`81`	`82`