fixing unit test

adtian2 · adtian2 · commit 848de311a863 · 2025-08-20T14:02:47.000-07:00
diff --git a/tests/unit/sagemaker/modules/train/sm_recipes/test_utils.py b/tests/unit/sagemaker/modules/train/sm_recipes/test_utils.py
@@ -240,7 +240,7 @@ def test_get_args_from_recipe_with_nova_and_role(mock_get_args_from_nova_recipe,
         {
             "model_type": "gpt_oss",
             "script": "custom_pretrain.py",
-            "model_base_name": "gpt_oss",
+            "model_base_name": "custom_model",
         },
     ],
 )

Original file line number	Diff line number	Diff line change
`@@ -240,7 +240,7 @@ def test_get_args_from_recipe_with_nova_and_role(mock_get_args_from_nova_recipe,`
`240`	`240`	`{`
`241`	`241`	`"model_type": "gpt_oss",`
`242`	`242`	`"script": "custom_pretrain.py",`
`243`		`- "model_base_name": "gpt_oss",`
	`243`	`+ "model_base_name": "custom_model",`
`244`	`244`	`},`
`245`	`245`	`],`
`246`	`246`	`)`