Use Any to use data.py

Author: ar90n

src/pytorch_lightning/tuner/batch_size_scaling.py

@@ -246,8 +246,9 @@ def _adjust_batch_size(
     if desc:
         log.info(f"Batch size {batch_size} {desc}, trying batch size {new_size}")
 
-    if not _is_valid_batch_size(new_size, trainer.train_dataloader, trainer):
-        new_size = min(new_size, len(trainer.train_dataloader.dataset))
+    train_dataloader: Any = trainer.train_dataloader
+    if not _is_valid_batch_size(new_size, train_dataloader, trainer):
+        new_size = min(new_size, len(train_dataloader.dataset))
 
     changed = new_size != batch_size
     lightning_setattr(model, batch_arg_name, new_size)