When overwriting the setup() function in the LightningDataModule, it is forced to add an unnecessary parameter.
🐛 Bug

When overwriting the setup() function in the LightningDataModule, it is forced to add an unnecessary parameter.

Environment

PyTorch Version (e.g., 1.0):1.7
OS (e.g., Linux):Windows
How you installed PyTorch (conda, pip, source):pip
Python version:3.6.12
CUDA/cuDNN version:11.0
GPU models and configuration:
Any other relevant information: def setup(self)

Additional context
I have to add another argument to setup(self) for it to work, such as setup(self,a), which I won't actually use at all.