Remove LightningDataModule from DataLoaderLike

borchero · Dec 26, 2021 · 82895c4 · 82895c4
1 parent 54aff76
commit 82895c4
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/lightkit/data/types.py b/lightkit/data/types.py
@@ -2,7 +2,6 @@
 import numpy as np
 import numpy.typing as npt
 import torch
-from pytorch_lightning import LightningDataModule
 from torch.utils.data import DataLoader, TensorDataset
 
 T = TypeVar("T")
@@ -18,7 +17,6 @@
 """
 
 DataLoaderLike = Union[
-    LightningDataModule,
     DataLoader[T],
     Sequence[DataLoader[T]],
 ]