diff --git a/dataset.py b/dataset.py index 99ad43e..866638e 100644 --- a/dataset.py +++ b/dataset.py @@ -66,6 +66,9 @@ def __getitem__(self, item): # print(data[0]) return torch.from_numpy(datapoint.astype('float32')) + def close(self): + pass + class OldH5Dataset(DepthDataset):