Skip to content

Commit dce06b1

Browse files
authored
fix: refresh MTRL attached-job test fixtures (#6259)
Update both MTRL attached-job integration paths to use a completed training job with an available model package. The previous fixture outputs no longer support the current evaluation flow. This restores attachment, fine-tuned evaluation, and metrics coverage.
1 parent 092ab8e commit dce06b1

2 files changed

Lines changed: 2 additions & 2 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
@@ -317,7 +317,7 @@ def test_evaluate_with_attached_trainer(self, lambda_agent_arn, test_config):
317317
from sagemaker.train.multi_turn_rl_trainer import MultiTurnRLTrainer
318318

319319
attached_job = MultiTurnRLTrainer.attach(
320-
"mock-oss-test-mtrl-20260616153024", session=boto3.Session(region_name=_REGION)
320+
"mock-oss-test-mtrl-20260910094327", session=boto3.Session(region_name=_REGION)
321321
)
322322

323323
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
@@ -54,7 +54,7 @@ def _get_account_id():
5454
"729646638167": {
5555
"env_name": "PROD",
5656
#"existing_job_name": "mock-oss-test-mtrl-20260611170946",
57-
"existing_job_name": "mock-oss-test-mtrl-20260806120845",
57+
"existing_job_name": "mock-oss-test-mtrl-20260910094327",
5858
"base_model": "mock-oss-test",
5959
"agent_core_arn": "arn:aws:bedrock-agentcore:us-west-2:729646638167:runtime/sagemaker_rft_prod_gsm8k_streaming-Yk6O377mUS",
6060
"dataset": "s3://sagemaker-rft-729646638167/prompts/gsm8k_small/prompts.parquet",

0 commit comments

Comments
 (0)