Skip to content

Commit 3dd7e63

Browse files
author
Roja Reddy Sareddy
committed
Update training resource name as previous MPG got cleanedup
1 parent 23784d9 commit 3dd7e63

3 files changed

Lines changed: 3 additions & 3 deletions

File tree

sagemaker-train/tests/integ/train/test_mtrl_evaluator_3p_agent.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -323,7 +323,7 @@ def test_evaluate_with_attached_trainer(self, lambda_agent_arn, test_config):
323323
from sagemaker.train.multi_turn_rl_trainer import MultiTurnRLTrainer
324324

325325
attached_job = MultiTurnRLTrainer.attach(
326-
"mock-oss-test-mtrl-20260615143910", session=boto3.Session(region_name=_REGION)
326+
"mock-oss-test-mtrl-20260616153024", session=boto3.Session(region_name=_REGION)
327327
)
328328

329329
evaluator = MultiTurnRLEvaluator(

sagemaker-train/tests/integ/train/test_mtrl_trainer_integration.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ def _get_account_id():
5353
"729646638167": {
5454
"env_name": "PROD",
5555
#"existing_job_name": "mock-oss-test-mtrl-20260611170946",
56-
"existing_job_name": "mock-oss-test-mtrl-20260615143910",
56+
"existing_job_name": "mock-oss-test-mtrl-20260616153024",
5757
"base_model": "mock-oss-test",
5858
"agent_core_arn": "arn:aws:bedrock-agentcore:us-west-2:729646638167:runtime/sagemaker_rft_prod_gsm8k_streaming-Yk6O377mUS",
5959
"dataset": "s3://sagemaker-rft-729646638167/prompts/gsm8k_small/prompts.parquet",

sagemaker-train/tests/integ/train/test_multi_turn_rl_trainer_integration.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ def _get_account_id():
4141
AGENT_RUNTIME_ID = "sagemaker_rft_prod_gsm8k_streaming-Yk6O377mUS"
4242
#BASE_MODEL = "openai-reasoning-gpt-oss-20b"
4343
BASE_MODEL = "mock-oss-test"
44-
EXISTING_JOB_NAME = "openai-reasoning-gpt-oss-20b-mtrl-20260602215955"
44+
EXISTING_JOB_NAME = "mock-oss-test-mtrl-20260616153024"
4545

4646

4747
@pytest.fixture(scope="module")

0 commit comments

Comments
 (0)