diff --git a/examples/pl_loops/kfold.py b/examples/pl_loops/kfold.py index af777df6f211d..38103d95ff3a6 100644 --- a/examples/pl_loops/kfold.py +++ b/examples/pl_loops/kfold.py @@ -152,12 +152,12 @@ def test_step(self, batch: Any, batch_idx: int, dataloader_idx: int = 0) -> None # self.reset(...) # # self.on_run_start(...) # # # -# while not self.done: # -# self.on_advance_start(...) # -# self.advance(...) # -# self.on_advance_end(...) # +# while not self.done: # +# self.on_advance_start(...) # +# self.advance(...) # +# self.on_advance_end(...) # # # -# return self.on_run_end(...) # +# return self.on_run_end(...) # #############################################################################################