Update test_dataloaders.py

Author: ananthsub

tests/trainer/test_dataloaders.py

@@ -756,14 +756,13 @@ class ModelWithDataLoaderDistributedSampler(EvalModelTemplate):
         def train_dataloader(self):
             dataloader = super().train_dataloader()
             dist_sampler = DistributedSampler(dataloader.dataset, shuffle=False)
-            dataloader = DataLoader(
+            return DataLoader(
                 dataset,
                 batch_size=self.batch_size,
                 drop_last=False,
                 sampler=dist_sampler,
                 shuffle=False
             )
-            return dataloader
 
     model = ModelWithDataLoaderDistributedSampler()
     trainer = Trainer(