Fixed mypy failure with DistributedSampler

pytorch · Jun 21, 2023 · 4f3b5f2 · 4f3b5f2
1 parent c54117e
commit 4f3b5f2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ignite/contrib/engines/common.py b/ignite/contrib/engines/common.py
@@ -264,7 +264,7 @@ def _setup_common_distrib_training_handlers(
 
         @trainer.on(Events.EPOCH_STARTED)
         def distrib_set_epoch(engine: Engine) -> None:
-            cast(DistributedSampler, train_sampler).set_epoch(engine.state.epoch - 1)
+            train_sampler.set_epoch(engine.state.epoch - 1)
 
 
 def empty_cuda_cache(_: Engine) -> None: