From 532a020d2af45290f2fea3ac3be521e373485b81 Mon Sep 17 00:00:00 2001 From: "Kim, Vinnam" Date: Thu, 30 Nov 2023 19:30:47 +0900 Subject: [PATCH] Fix wrong docstrings Signed-off-by: Kim, Vinnam --- src/otx/algo/callbacks/iteration_timer.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/otx/algo/callbacks/iteration_timer.py b/src/otx/algo/callbacks/iteration_timer.py index 5fc14fc8609..da1f0b65bbf 100644 --- a/src/otx/algo/callbacks/iteration_timer.py +++ b/src/otx/algo/callbacks/iteration_timer.py @@ -87,7 +87,7 @@ def on_train_batch_end( batch: Any, # noqa: ANN401 batch_idx: int, ) -> None: - """Log iteration time on the training batch start.""" + """Log iteration time on the training batch end.""" self._on_batch_end(pl_module=pl_module, phase="train") def on_validation_batch_start( @@ -110,7 +110,7 @@ def on_validation_batch_end( batch_idx: int, dataloader_idx: int = 0, ) -> None: - """Log iteration time on the validation batch start.""" + """Log iteration time on the validation batch end.""" self._on_batch_end(pl_module=pl_module, phase="validation") def on_test_batch_start( @@ -133,5 +133,5 @@ def on_test_batch_end( batch_idx: int, dataloader_idx: int = 0, ) -> None: - """Log iteration time on the test batch start.""" + """Log iteration time on the test batch end.""" self._on_batch_end(pl_module=pl_module, phase="test")