Update DB classname

midusi · Aug 1, 2022 · f8e8864 · f8e8864
1 parent c9f8b2b
commit f8e8864
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lsat/dataset/LSA_Dataset.py → lsat/dataset/PyTorchDataset.py b/lsat/dataset/LSA_Dataset.py → lsat/dataset/PyTorchDataset.py
@@ -33,7 +33,7 @@ def _load_clip_as_tensors(paths: dict[str, Path]) -> Iterable[Tensor]:
     return (map(lambda frame: frame['data'], VideoReader(str(paths['mp4']), "video")))
 
 
-class LSA_Dataset(Dataset):
+class PyTorchDataset(Dataset):
 
     def __init__(self,
             root: str,