Update src/lightning/pytorch/trainer/connectors/data_connector.py

Co-authored-by: Jirka Borovec <[email protected]>

Author: ved1beta

src/lightning/pytorch/trainer/connectors/data_connector.py

@@ -285,7 +285,7 @@ def __len__(self):
                 sampler = SequentialSampler(dl.dataset)
 
             # Create a new dataloader with the new sampler
-            new_dl = DataLoader(
+            dl = DataLoader(
                 dataset=dl.dataset,
                 batch_size=dl.batch_size,
                 sampler=sampler,
@@ -300,9 +300,7 @@ def __len__(self):
                 prefetch_factor=dl.prefetch_factor,
                 persistent_workers=dl.persistent_workers,
             )
-            updated.append(new_dl)
-        else:
-            updated.append(dl)
+        updated.append(dl)
 
     combined_loader.flattened = updated