From e190bc5f93818e2bb26825511e2d1c0a97978d40 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Wed, 18 Apr 2018 14:21:54 +0200
Subject: k

---
 test.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'test.py')

diff --git a/test.py b/test.py
index 74768d2..f6e106b 100644
--- a/test.py
+++ b/test.py
@@ -17,6 +17,7 @@ if __name__ == '__main__':
 
     data_loader = CreateRecursiveDataLoader(opt)
     dataset = data_loader.load_data()
+    ds = dataset.dataset
     model = create_model(opt)
     visualizer = Visualizer(opt)
     # create website
@@ -39,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.dataset.append(next_image)
+            ds.append(next_image)
 
     webpage.save()
-- 
cgit v1.2.3-70-g09d2