<filename>src/lit_tracking/datamodule.py
from pytorch_lightning import LightningDataModule


class TrackingDataModule(LightningDataModule):
    def __init__(self):
        super().__init__()

<filename>src/lit_tracking/datamodule.py
from pytorch_lightning import LightningDataModule


class TrackingDataModule(LightningDataModule):
    def __init__(self):
        super().__init__()
