Protocols Module # Bases: Protocol[P, OutT] Small protocol used to help annotate the input/outputs of torch.nn.Modules. DataModule # Bases: Protocol[BatchType] Protocol that shows the minimal attributes / methods of the LightningDataModule class. This is used to type hint the batches that are yielded by the DataLoaders.