From 88976997040c542f874cdaf1d67bb7c47e96a536 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Sun, 17 Jun 2018 09:22:10 +0200
Subject: args

---
 data/recursive_dataset.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'data/recursive_dataset.py')

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
-- 
cgit v1.2.3-70-g09d2