k

2 files changed, 2 insertions, 2 deletions

diff --git a/data/__init__.py b/data/__init__.py
index 6da79f3..cd868fc 100644
--- a/data/__init__.py
+++ b/data/__init__.py
@@ -64,7 +64,7 @@ class CustomDatasetDataLoader(BaseDataLoader):
 
 class RecursiveDatasetDataLoader(BaseDataLoader):
     def name(self):
-        return 'CustomDatasetDataLoader'
+        return 'RecursiveDatasetDataLoader'
 
     def initialize(self, opt):
         #BaseDataLoader.initialize(self, opt)
diff --git a/test.py b/test.py
index 53a1d4b..ad42b2d 100644
--- a/test.py
+++ b/test.py
@@ -32,7 +32,7 @@ if __name__ == '__main__':
         print('%04d: process image... %s' % (i, img_path))
         ims = visualizer.save_images(webpage, visuals, img_path, aspect_ratio=opt.aspect_ratio)
         print(dataset.name())
-        if dataset.name() == 'RecursiveImageDataset':
+        if dataset.name() == 'RecursiveDatasetDataLoader':
             # dataset.append(ims)
             print(ims)