[Train] Report only once in SklearnTrainer (ray-project#30593)

Sets "done": True in the SklearnTrainer report so that it only print the results only once. Signed-off-by: Antoni Baum <antoni.baum@protonmail.com> Signed-off-by: Weichen Xu <weichen.xu@databricks.com>
WeichenXu123 · Dec 19, 2022 · b604008 · b604008
1 parent eb317a6
commit b604008
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/ray/train/sklearn/sklearn_trainer.py b/python/ray/train/sklearn/sklearn_trainer.py
@@ -433,5 +433,6 @@ def training_loop(self) -> None:
             **validation_set_scores,
             **cv_scores,
             "fit_time": fit_time,
+            "done": True,
         }
         self._report(results)