explicit model argument

datasets/shapenet.py

@@ -1,4 +1,5 @@
 from pathlib import Path
+from warnings import warn
 
 import numpy as np
 
@@ -55,13 +56,14 @@ class CustomShapeNet(InMemoryDataset):
     def processed_file_names(self):
         return [f'{self.mode}.pt']
 
-    def download(self):
+    def __download(self):
         dir_count = len([name for name in os.listdir(self.raw_dir) if os.path.isdir(os.path.join(self.raw_dir, name))])
 
         if dir_count:
             print(f'{dir_count} folders have been found....')
             return dir_count
-        raise IOError("No raw pointclouds have been found.")
+        warn(ResourceWarning("No raw pointclouds have been found. Was this intentional?"))
+        return dir_count
 
     @property
     def num_classes(self):