diff --git a/replicate/training.py b/replicate/training.py index 28e28b4..60fbf82 100644 --- a/replicate/training.py +++ b/replicate/training.py @@ -61,6 +61,9 @@ class Training(Resource): logs: Optional[str] """The logs of the training.""" + metrics: Optional[Dict[str, Any]] + """Metrics for the training.""" + error: Optional[str] """The error encountered during the training, if any."""