debuggin..

1 files changed, 4 insertions, 3 deletions

diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py
index b6ce65d..1ec5e19 100644
--- a/data/recursive_dataset.py
+++ b/data/recursive_dataset.py
@@ -4,7 +4,8 @@ from data.base_dataset import BaseDataset, get_transform
 from data.image_folder import make_dataset
 from PIL import Image
 import random
-import gevent
+# import gevent
+import time
 
 class RecursiveDataset(BaseDataset):
     def initialize(self, opt):
@@ -26,9 +27,9 @@ class RecursiveDataset(BaseDataset):
                 A_path = A_path2
             else:
                 print('not there yet... {}'.format(A_path))
-                print('sleeping for {}'.format(self.opt.poll_delay))
                 while not os.path.exists(A_path):
-                    gevent.sleep(self.opt.poll_delay)
+                    print('sleeping for {}'.format(self.opt.poll_delay))
+                    time.sleep(self.opt.poll_delay)
         print("got {}".format(A_path))
         A_img = Image.open(A_path).convert('RGB')
         A = self.transform(A_img)