diff options
Diffstat (limited to 'data/recursive_dataset.py')
| -rw-r--r-- | data/recursive_dataset.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py index 84a6307..b399909 100644 --- a/data/recursive_dataset.py +++ b/data/recursive_dataset.py @@ -31,6 +31,9 @@ class RecursiveDataset(BaseDataset): return {'A': A, 'A_paths': A_path} + def load_data(self): + return self + def __len__(self): return float("inf") |