From 20f218b40b94bc2962dadc79865f2cec29fe37d2 Mon Sep 17 00:00:00 2001
From: Roja Reddy Sareddy <rsareddy@amazon.com>
Date: Thu, 23 Jul 2026 14:30:46 -0700
Subject: [PATCH] Fix gpu integ test failure due to outdated MPG

---
 .../tests/integ/train/test_mtrl_trainer_integration.py          | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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 9f30c3c86b..68dafd9d2f 100644
--- a/sagemaker-train/tests/integ/train/test_mtrl_trainer_integration.py
+++ b/sagemaker-train/tests/integ/train/test_mtrl_trainer_integration.py
@@ -60,7 +60,7 @@ def _get_account_id():
         "dataset": "s3://sagemaker-rft-729646638167/prompts/gsm8k_small/prompts.parquet",
         "s3_output_path": "s3://sagemaker-us-west-2-729646638167/mtrl-integ/eval-output/",
         "mlflow_resource_arn": "arn:aws:sagemaker:us-west-2:729646638167:mlflow-app/app-TTAUWUNMUHH6",
-        "model_package_group": "arn:aws:sagemaker:us-west-2:729646638167:model-package-group/openai-reasoning-gpt-oss-20b-mtrl-mpg",
+        "model_package_group": "arn:aws:sagemaker:us-west-2:729646638167:model-package-group/mock-oss-test-mtrl-mpg",
     },
     # PREPROD — Staging account (391266019386)
     "391266019386": {