Skip to content

Commit 696257b

Browse files
committed
ci: add docstring to describe new test
1 parent b91a372 commit 696257b

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/tests_pytorch/loops/test_trainer_iterable_dataset_double_iter.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,10 @@ def train_model(queue, maxEpochs, ckptPath):
3737
return trainer
3838

3939
def test_training():
40+
"""
41+
Test that reproduces issue in calling iter twice on a queue-based
42+
IterableDataset leads to Queue Empty errors when resuming from a checkpoint.
43+
"""
4044
queue = create_queue()
4145

4246
ckpt_path = "model.ckpt"

0 commit comments

Comments
 (0)