Skip to content

Commit 60f12e3

Browse files
committed
Remove unintentional dict's items() call
1 parent 4e2be2f commit 60f12e3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytorch_lightning/utilities/auto_restart.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -344,7 +344,7 @@ def _dataloader_to_state_dict(
344344
if not isinstance(dataloader.dataset, CaptureIterableDataset):
345345
fast_forward_sampler = _find_fast_forward_samplers(dataloader)
346346
if fast_forward_sampler is not None:
347-
samplers_state_dict = fast_forward_sampler.state_dict(num_batches_processed=num_batches_processed).items()
347+
samplers_state_dict = fast_forward_sampler.state_dict(num_batches_processed=num_batches_processed)
348348
for sampler_k, sampler_v in samplers_state_dict.items():
349349
out[sampler_k] = sampler_v
350350
return out

0 commit comments

Comments
 (0)