From 919ba03dae6ca25efc2f0bafc5a027efc0fe18e1 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Wed, 18 Apr 2018 14:25:28 +0200
Subject: k

---
 data/__init__.py | 3 +++
 test.py          | 2 +-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/data/__init__.py b/data/__init__.py
index 0c7a752..9d34e6f 100644
--- a/data/__init__.py
+++ b/data/__init__.py
@@ -76,6 +76,9 @@ class RecursiveDatasetDataLoader(BaseDataLoader):
             shuffle=not opt.serial_batches,
             num_workers=int(opt.nThreads))
 
+    def append(self, arg):
+        self.dataset.append(arg)
+
     def load_data(self):
         return self
 
diff --git a/test.py b/test.py
index f6e106b..066c6c1 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()
-            ds.append(next_image)
+            dataset.append(next_image)
 
     webpage.save()
-- 
cgit v1.2.3-70-g09d2