Skip to content

Commit 81d002d

Browse files
committed
if data is none then batches are not run
1 parent 6c5ad29 commit 81d002d

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

tests/ignite/engine/test_engine.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -645,10 +645,16 @@ def _test_check_triggered_events(
645645
n_iter_completed = max_epochs * epoch_length if max_epochs is not None else max_iters
646646

647647
if n_batch_started is None:
648-
n_batch_started = max_epochs * epoch_length if max_epochs is not None else max_iters
648+
if data is None:
649+
n_batch_started = 0
650+
else:
651+
n_batch_started = max_epochs * epoch_length if max_epochs is not None else max_iters
649652

650653
if n_batch_completed is None:
651-
n_batch_completed = max_epochs * epoch_length if max_epochs is not None else max_iters
654+
if data is None:
655+
n_batch_completed = 0
656+
else:
657+
n_batch_completed = max_epochs * epoch_length if max_epochs is not None else max_iters
652658

653659
if n_terminate is None:
654660
n_terminate = int(n_epoch_started != n_epoch_completed) if max_iters is not None else 0

0 commit comments

Comments
 (0)