Skip to content

Commit

Permalink
Fix wrong docstrings
Browse files Browse the repository at this point in the history
Signed-off-by: Kim, Vinnam <vinnam.kim@intel.com>
  • Loading branch information
vinnamkim committed Nov 30, 2023
1 parent 2a9313e commit 532a020
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions src/otx/algo/callbacks/iteration_timer.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand All @@ -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(
Expand All @@ -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")

0 comments on commit 532a020

Please sign in to comment.