Typing utils
Utilities to help annotate the types of values in the project.
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.
Module #
Bases: Protocol[P, OutT]
Small protocol used to help annotate the input/outputs of torch.nn.Module
s.
is_sequence_of #
Used to check (and tell the type checker) that object
is a sequence of items of this
type.