From 146014b8e7ff0e9fd1020e62cabcce471f988728 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Wed, 18 Apr 2018 13:16:24 +0200 Subject: k --- data/recursive_dataset.py | 3 +++ 1 file changed, 3 insertions(+) (limited to 'data/recursive_dataset.py') 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") -- cgit v1.2.3-70-g09d2