File tree Expand file tree Collapse file tree
sagemaker-train/recipe_tests Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1212# language governing permissions and limitations under the License.
1313"""Recipe validation integ test for the HP-ModelCustomization-RecipeValidator pipeline.
1414
15- Iterates through every model in the private hub referenced by the ``HYPERPOD_HUB_NAME ``
15+ Iterates through every model in the private hub referenced by the ``SAGEMAKER_HUB_NAME ``
1616env var and validates that each fine-tuning recipe can be used to instantiate the
1717appropriate ``sagemaker.train`` Trainer class (SFT/DPO/RLAIF/RLVR).
1818"""
@@ -66,8 +66,8 @@ def detect_lora_or_full(recipe_path: str) -> TrainingType:
6666
6767def test_new_recipes_create_valid_trainers ():
6868 """Validate every new/modified recipe in the private hub yields a valid Trainer."""
69- hub_name = os .environ .get ("HYPERPOD_HUB_NAME " )
70- assert hub_name , "HYPERPOD_HUB_NAME environment variable must be set"
69+ hub_name = os .environ .get ("SAGEMAKER_HUB_NAME " )
70+ assert hub_name , "SAGEMAKER_HUB_NAME environment variable must be set"
7171
7272 sm = boto3 .client ("sagemaker" , region_name = "us-west-2" )
7373
You can’t perform that action at this time.
0 commit comments