Skip to content

Commit 76c5ce1

Browse files
authored
[Fix] Fix after iter SegVisualizationHook (#1908)
* [Fix] Fix after iter `SegVisualizationHook` * ut
1 parent c498e6f commit 76c5ce1

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

mmseg/engine/hooks/visualization_hook.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -62,12 +62,12 @@ def __init__(self,
6262
'effect. The results will NOT be '
6363
'visualized or stored.')
6464

65-
def after_iter(self,
66-
runner: Runner,
67-
batch_idx: int,
68-
data_batch: Sequence[dict],
69-
outputs: Sequence[SegDataSample],
70-
mode: str = 'val') -> None:
65+
def _after_iter(self,
66+
runner: Runner,
67+
batch_idx: int,
68+
data_batch: Sequence[dict],
69+
outputs: Sequence[SegDataSample],
70+
mode: str = 'val') -> None:
7171
"""Run after every ``self.interval`` validation iterations.
7272
7373
Args:

tests/test_engine/test_visualization_hook.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,4 +59,4 @@ def test_after_test_iter(self):
5959
runner = Mock()
6060
runner.iter = 3
6161
hook = SegVisualizationHook(draw=True, interval=1)
62-
hook.after_iter(runner, 1, self.data_batch, self.outputs)
62+
hook.after_test_iter(runner, 1, self.data_batch, self.outputs)

0 commit comments

Comments
 (0)