summaryrefslogtreecommitdiff
path: root/data
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2018-06-17 09:22:10 +0200
committerJules Laplace <julescarbon@gmail.com>2018-06-17 09:22:10 +0200
commit88976997040c542f874cdaf1d67bb7c47e96a536 (patch)
tree08fc9f53f96b8f8ec1adf0f4066b3018d507b3c9 /data
parent9467998d18246acb60d76349826b2e8ebaedfe9c (diff)
args
Diffstat (limited to 'data')
-rw-r--r--data/recursive_dataset.py2
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