k

2 files changed, 1 insertions, 4 deletions

diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py
index b655147..15151d9 100644
--- a/data/recursive_dataset.py
+++ b/data/recursive_dataset.py
@@ -31,9 +31,6 @@ class RecursiveDataset(BaseDataset):
 
         return {'A': A, 'A_paths': A_path}
 
-    def load_data(self):
-        return self
-
     def __len__(self):
         return 10000000
 
diff --git a/test.py b/test.py
index b2c679c..1328f18 100644
--- a/test.py
+++ b/test.py
@@ -40,6 +40,6 @@ if __name__ == '__main__':
             cmd = ("/bin/cp", last_image, next_image)
             process = subprocess.Popen(cmd, stdout=subprocess.PIPE)
             output, error = process.communicate()
-            dataset.append(next_image)
+            dataset.dataset.append(next_image)
 
     webpage.save()