Skip to content

Commit 54d1618

Browse files
committed
Update env var to SAGEMAKER_HUB_NAME
1 parent 3c52524 commit 54d1618

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

sagemaker-train/recipe_tests/test_recipe_validation.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
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``
1616
env var and validates that each fine-tuning recipe can be used to instantiate the
1717
appropriate ``sagemaker.train`` Trainer class (SFT/DPO/RLAIF/RLVR).
1818
"""
@@ -66,8 +66,8 @@ def detect_lora_or_full(recipe_path: str) -> TrainingType:
6666

6767
def 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

0 commit comments

Comments
 (0)