Skip to content

Commit 3ae0d3a

Browse files
author
Chad Chiang
committed
solve another unit test error
1 parent 395a435 commit 3ae0d3a

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/sagemaker/modules/train/model_trainer.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -240,6 +240,7 @@ class ModelTrainer(BaseModel):
240240
_infra_check_config: Optional[InfraCheckConfig] = PrivateAttr(default=None)
241241
_session_chaining_config: Optional[SessionChainingConfig] = PrivateAttr(default=None)
242242
_remote_debug_config: Optional[RemoteDebugConfig] = PrivateAttr(default=None)
243+
_metric_definitions: Optional[List[MetricDefinition]] = PrivateAttr(default=None)
243244

244245
_temp_recipe_train_dir: Optional[TemporaryDirectory] = PrivateAttr(default=None)
245246

@@ -697,6 +698,7 @@ def train(
697698
training_image_config=self.training_image_config,
698699
container_entrypoint=container_entrypoint,
699700
container_arguments=container_arguments,
701+
metric_definitions=self._metric_definitions,
700702
)
701703

702704
resource_config = self.compute._to_resource_config()

0 commit comments

Comments
 (0)