k

1 files changed, 4 insertions, 2 deletions

diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py
index b20b082..8b79d9b 100644
--- a/data/recursive_dataset.py
+++ b/data/recursive_dataset.py
@@ -2,15 +2,17 @@ import os.path
 from data.base_dataset import BaseDataset, get_transform
 from data.image_folder import make_dataset
 from PIL import Image
+import random
 
 class NextImage():
     def __init__(self):
         self.next_image = "lol"
+        self.id = str(random.randint(1,100))
     def append(self, path):
         self.next_image = path
-        print("APPEND " + self.next_image)
+        print("APPEND " + self.id + ' ' + self.next_image)
     def get(self):
-        print("GET " + self.next_image)
+        print("GET " + self.id + ' ' + self.next_image)
         return self.next_image
 
 im = NextImage()