diff --git a/sagemaker-train/tests/integ/train/test_mtrl_evaluator_3p_agent.py b/sagemaker-train/tests/integ/train/test_mtrl_evaluator_3p_agent.py index 1a9d1402c3..6c2da654cf 100644 --- a/sagemaker-train/tests/integ/train/test_mtrl_evaluator_3p_agent.py +++ b/sagemaker-train/tests/integ/train/test_mtrl_evaluator_3p_agent.py @@ -317,7 +317,7 @@ def test_evaluate_with_attached_trainer(self, lambda_agent_arn, test_config): from sagemaker.train.multi_turn_rl_trainer import MultiTurnRLTrainer attached_job = MultiTurnRLTrainer.attach( - "mock-oss-test-mtrl-20260616153024", session=boto3.Session(region_name=_REGION) + "mock-oss-test-mtrl-20260910094327", session=boto3.Session(region_name=_REGION) ) evaluator = MultiTurnRLEvaluator( diff --git a/sagemaker-train/tests/integ/train/test_mtrl_trainer_integration.py b/sagemaker-train/tests/integ/train/test_mtrl_trainer_integration.py index 1f506560b7..65a07ce784 100644 --- a/sagemaker-train/tests/integ/train/test_mtrl_trainer_integration.py +++ b/sagemaker-train/tests/integ/train/test_mtrl_trainer_integration.py @@ -54,7 +54,7 @@ def _get_account_id(): "729646638167": { "env_name": "PROD", #"existing_job_name": "mock-oss-test-mtrl-20260611170946", - "existing_job_name": "mock-oss-test-mtrl-20260806120845", + "existing_job_name": "mock-oss-test-mtrl-20260910094327", "base_model": "mock-oss-test", "agent_core_arn": "arn:aws:bedrock-agentcore:us-west-2:729646638167:runtime/sagemaker_rft_prod_gsm8k_streaming-Yk6O377mUS", "dataset": "s3://sagemaker-rft-729646638167/prompts/gsm8k_small/prompts.parquet",