From c235722b7fdce8d9d165c39dba044743505765ac Mon Sep 17 00:00:00 2001
From: aviruthen <91846056+aviruthen@users.noreply.github.com>
Date: Thu, 8 Jan 2026 17:56:48 -0800
Subject: [PATCH] Remove experiment_config parameter

---
 .../sagemaker/train/aws_batch/training_queue.py  | 16 ----------------
 1 file changed, 16 deletions(-)

diff --git a/sagemaker-train/src/sagemaker/train/aws_batch/training_queue.py b/sagemaker-train/src/sagemaker/train/aws_batch/training_queue.py
index f627d41026..bdcc54ad4f 100644
--- a/sagemaker-train/src/sagemaker/train/aws_batch/training_queue.py
+++ b/sagemaker-train/src/sagemaker/train/aws_batch/training_queue.py
@@ -41,7 +41,6 @@ def submit(
         share_identifier: Optional[str] = None,
         timeout: Optional[Dict] = None,
         tags: Optional[Dict] = None,
-        experiment_config: Optional[Dict] = None,
     ) -> TrainingQueuedJob:
         """Submit a queued job and return a QueuedJob object.
 
@@ -54,9 +53,6 @@ def submit(
             share_identifier: Share identifier for Batch job.
             timeout: Timeout configuration for Batch job.
             tags: Tags apply to Batch job. These tags are for Batch job only.
-            experiment_config: Experiment management configuration.
-                Optionally, the dict can contain four keys:
-                'ExperimentName', 'TrialName', 'TrialComponentDisplayName' and 'RunName'.
 
         Returns: a TrainingQueuedJob object with Batch job ARN and job name.
 
@@ -71,11 +67,6 @@ def submit(
             raise ValueError(
                 "TrainingQueue requires using a ModelTrainer with Mode.SAGEMAKER_TRAINING_JOB"
             )
-        if experiment_config is not None:
-            logging.warning(
-                "ExperimentConfig is not supported for ModelTrainer. "
-                "It will be ignored when submitting the job."
-            )
         training_payload = training_job._create_training_job_args(
             input_data_config=inputs, boto3=True
         )
@@ -111,7 +102,6 @@ def map(
         share_identifier: Optional[str] = None,
         timeout: Optional[Dict] = None,
         tags: Optional[Dict] = None,
-        experiment_config: Optional[Dict] = None,
     ) -> List[TrainingQueuedJob]:
         """Submit queued jobs to the provided estimator and return a list of TrainingQueuedJob objects.
 
@@ -124,15 +114,10 @@ def map(
             share_identifier: Share identifier for the Batch jobs.
             timeout: Timeout configuration for the Batch jobs.
             tags: Tags apply to Batch job. These tags are for Batch job only.
-            experiment_config: Experiment management configuration.
-                Optionally, the dict can contain four keys:
-                'ExperimentName', 'TrialName', 'TrialComponentDisplayName' and 'RunName'.
 
         Returns: a list of TrainingQueuedJob objects with each Batch job ARN and job name.
 
         """
-        if experiment_config is None:
-            experiment_config = {}
 
         if job_names is not None:
             if len(job_names) != len(inputs):
@@ -153,7 +138,6 @@ def map(
                 share_identifier,
                 timeout,
                 tags,
-                experiment_config,
             )
             queued_batch_job_list.append(queued_batch_job)