Modify Workflow to Allow IterableDataset Inputs

Signed-off-by: Eric Kerfoot <[email protected]>

Author: ericspod

monai/engines/workflow.py

@@ -128,6 +128,7 @@ def __init__(
 
             # set the epoch value for DistributedSampler objects when an epoch starts
             if isinstance(sampler, DistributedSampler):
+
                 @self.on(Events.EPOCH_STARTED)
                 def set_sampler_epoch(engine: Engine) -> None:
                     sampler.set_epoch(engine.state.epoch)

tests/test_iterable_dataset.py

@@ -18,12 +18,11 @@
 
 import nibabel as nib
 import numpy as np
+import torch.nn as nn
 
 from monai.data import DataLoader, Dataset, IterableDataset
-from monai.transforms import Compose, LoadImaged, SimulateDelayd
 from monai.engines import SupervisedEvaluator
-
-import torch.nn as nn
+from monai.transforms import Compose, LoadImaged, SimulateDelayd
 
 
 class _Stream: