diff options
Diffstat (limited to 'data/recursive_dataset.py')
| -rw-r--r-- | data/recursive_dataset.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py index 87f331c..728bfba 100644 --- a/data/recursive_dataset.py +++ b/data/recursive_dataset.py @@ -35,7 +35,7 @@ class RecursiveDataset(BaseDataset): return input_dict def __len__(self): - return len(self.A_paths) + return self.dataset_size def name(self): return 'RecursiveDataset'
\ No newline at end of file |