diff --git a/sagemaker-train/tests/integ/train/test_cpt_data_mixing_hyperpod.py b/sagemaker-train/tests/integ/train/test_cpt_data_mixing_hyperpod.py index 64c12689cc..249aba5652 100644 --- a/sagemaker-train/tests/integ/train/test_cpt_data_mixing_hyperpod.py +++ b/sagemaker-train/tests/integ/train/test_cpt_data_mixing_hyperpod.py @@ -29,6 +29,7 @@ import json import logging +import os import subprocess import time import random @@ -46,7 +47,9 @@ # Test configuration REGION = "us-east-1" -CLUSTER_NAME = "riv-rig" +# HyperPod cluster provisioned for the pysdk integ-test account in us-east-1. +# Overridable via env var so a cluster rename does not require a code change. +CLUSTER_NAME = os.environ.get("PYSDK_HYPERPOD_CLUSTER_NAME", "pysdk-hp-integ-tests") INSTANCE_TYPE = "ml.p5.48xlarge" NODE_COUNT = 2 MODEL_NAME = "nova-textgeneration-micro" diff --git a/sagemaker-train/tests/integ/train/test_cpt_hyperpod.py b/sagemaker-train/tests/integ/train/test_cpt_hyperpod.py index 0bf31e4ab1..2a10a732e9 100644 --- a/sagemaker-train/tests/integ/train/test_cpt_hyperpod.py +++ b/sagemaker-train/tests/integ/train/test_cpt_hyperpod.py @@ -29,6 +29,7 @@ import json import logging +import os import subprocess import time import random @@ -45,7 +46,9 @@ # Test configuration REGION = "us-east-1" -CLUSTER_NAME = "riv-rig" +# HyperPod cluster provisioned for the pysdk integ-test account in us-east-1. +# Overridable via env var so a cluster rename does not require a code change. +CLUSTER_NAME = os.environ.get("PYSDK_HYPERPOD_CLUSTER_NAME", "pysdk-hp-integ-tests") INSTANCE_TYPE = "ml.p5.48xlarge" NODE_COUNT = 2 MODEL_NAME = "nova-textgeneration-micro" diff --git a/sagemaker-train/tests/integ/train/test_sft_data_mixing_hyperpod.py b/sagemaker-train/tests/integ/train/test_sft_data_mixing_hyperpod.py index 5e7c00b045..c9668fa421 100644 --- a/sagemaker-train/tests/integ/train/test_sft_data_mixing_hyperpod.py +++ b/sagemaker-train/tests/integ/train/test_sft_data_mixing_hyperpod.py @@ -30,6 +30,7 @@ import json import logging +import os import time import random @@ -49,7 +50,9 @@ REGION = "us-east-1" DATA_PREFIX = "test-sft-data-mixing-hyperpod-integ" NUM_TRAINING_SAMPLES = 300 -HYPERPOD_CLUSTER_NAME = "riv-rig" +# HyperPod cluster provisioned for the pysdk integ-test account in us-east-1. +# Overridable via env var so a cluster rename does not require a code change. +HYPERPOD_CLUSTER_NAME = os.environ.get("PYSDK_HYPERPOD_CLUSTER_NAME", "pysdk-hp-integ-tests") HYPERPOD_INSTANCE_TYPE = "ml.p5.48xlarge"