fix merge artifact

benieric · benieric · commit c5b1e12f25df · 2025-03-04T18:40:49.000-08:00
diff --git a/src/sagemaker/modules/distributed.py b/src/sagemaker/modules/distributed.py
@@ -76,7 +76,7 @@ def _to_mp_hyperparameters(self) -> Dict[str, Any]:
         return hyperparameters
 
 
-class DistributedConfig(BaseModel, ABC):
+class DistributedConfig(BaseConfig, ABC):
     """Abstract base class for distributed training configurations.
 
     This class defines the interface that all distributed training configurations
diff --git a/tests/integ/sagemaker/modules/train/test_model_trainer.py b/tests/integ/sagemaker/modules/train/test_model_trainer.py
@@ -142,7 +142,6 @@ def driver_script(self) -> str:
 
     source_code = SourceCode(
         source_dir=f"{DATA_DIR}/modules/scripts",
-        requirements=
         entry_script="entry_script.py",
     )
 
@@ -158,3 +157,4 @@ def driver_script(self) -> str:
         distributed=custom_driver,
         base_job_name="custom-distributed-driver",
     )
+    model_trainer.train()

Original file line number	Diff line number	Diff line change
`@@ -142,7 +142,6 @@ def driver_script(self) -> str:`
`142`	`142`
`143`	`143`	`source_code = SourceCode(`
`144`	`144`	`source_dir=f"{DATA_DIR}/modules/scripts",`
`145`		`- requirements=`
`146`	`145`	`entry_script="entry_script.py",`
`147`	`146`	`)`
`148`	`147`
`@@ -158,3 +157,4 @@ def driver_script(self) -> str:`
`158`	`157`	`distributed=custom_driver,`
`159`	`158`	`base_job_name="custom-distributed-driver",`
`160`	`159`	`)`
	`160`	`+ model_trainer.train()`