Skip to content

Commit 3fd063c

Browse files
author
Chad Chiang
committed
style fix
1 parent a0f1f2b commit 3fd063c

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/sagemaker/modules/train/model_trainer.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1291,12 +1291,14 @@ def with_checkpoint_config(
12911291
"""
12921292
self.checkpoint_config = checkpoint_config or configs.CheckpointConfig()
12931293
return self
1294-
1294+
12951295
def with_metric_definitions(
12961296
self, metric_definitions: List[MetricDefinition]
12971297
) -> "ModelTrainer": # noqa: D412
12981298
"""Set the metric definitions for the training job.
1299+
12991300
Example:
1301+
13001302
.. code:: python
13011303
from sagemaker.modules.train import ModelTrainer
13021304
from sagemaker.modules.configs import MetricDefinition
@@ -1309,6 +1311,7 @@ def with_metric_definitions(
13091311
model_trainer = ModelTrainer(
13101312
...
13111313
).with_metric_definitions(metric_definitions)
1314+
13121315
Args:
13131316
metric_definitions (List[MetricDefinition]):
13141317
The metric definitions for the training job.

0 commit comments

Comments
 (0)