k

1 files changed, 2 insertions, 2 deletions

diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py
index 5cb0969..7d1db84 100644
--- a/data/recursive_dataset.py
+++ b/data/recursive_dataset.py
@@ -11,7 +11,7 @@ class NextImage():
     def append(self, path):
         self.next_image = path
         print("APPEND " + self.id + ' ' + self.next_image)
-    def get(self):
+    def get(self, index):
         print("GET " + self.id + ' ' + self.next_image)
         return self.next_image
 
@@ -34,7 +34,7 @@ class RecursiveDataset(BaseDataset):
         self.transform = get_transform(opt)
 
     def __getitem__(self, index):
-        A_path = im.get()
+        A_path = im.get(index)
         print('next image: ' + A_path)
         A_img = Image.open(A_path).convert('RGB')
         A = self.transform(A_img)