fix: aws_batch/test_training_queue QM unit test fix

davlind-amzn · davlind-amzn · commit 3dc14b90e4b8 · 2026-03-21T09:26:48.000-07:00
diff --git a/sagemaker-train/tests/integ/train/aws_batch/manager.py b/sagemaker-train/tests/integ/train/aws_batch/manager.py
@@ -21,9 +21,9 @@ class BatchTestResourceManager:
     def __init__(
         self,
         batch_client,
-        queue_name="pysdk-test-queue",
-        service_env_name="pysdk-test-queue-service-environment",
-        scheduling_policy_name="pysdk-test-scheduling-policy",
+        queue_name="pysdk-test-qm-queue",
+        service_env_name="pysdk-test-qm-queue-service-environment",
+        scheduling_policy_name="pysdk-test-qm-scheduling-policy",
         quota_share_name="pysdk-test-quota-share",
     ):
         self.batch_client = batch_client
diff --git a/sagemaker-train/tests/unit/train/aws_batch/test_training_queue.py b/sagemaker-train/tests/unit/train/aws_batch/test_training_queue.py
@@ -403,7 +403,6 @@ def test_submit_with_quota_share_name_and_preemption_config(self, mock_submit_se
             SHARE_IDENTIFIER,
             TIMEOUT_CONFIG,
             BATCH_TAGS,
-            None,
             quota_share_name=QUOTA_SHARE_NAME,
             preemption_config=PREEMPTION_CONFIG,
         )

Original file line number	Diff line number	Diff line change
`@@ -403,7 +403,6 @@ def test_submit_with_quota_share_name_and_preemption_config(self, mock_submit_se`
`403`	`403`	`SHARE_IDENTIFIER,`
`404`	`404`	`TIMEOUT_CONFIG,`
`405`	`405`	`BATCH_TAGS,`
`406`		`- None,`
`407`	`406`	`quota_share_name=QUOTA_SHARE_NAME,`
`408`	`407`	`preemption_config=PREEMPTION_CONFIG,`
`409`	`408`	`)`