dists for windows

mattclifford1 · Jul 17, 2024 · ad79e31 · ad79e31
1 parent e3dc71f
commit ad79e31
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/IQM_Vis/examples/dists.py b/IQM_Vis/examples/dists.py
@@ -92,7 +92,9 @@ def run():
     metric_images = {}
 
     # make dataset list of images
-    files = sorted(glob.glob('/home/matt/datasets/Textures/*'))
+    files = sorted(glob.glob(os.path.join(os.path.expanduser('~'), 'datasets', 'Textures', '*')))
+    if len(files) == 0:
+        raise ValueError('No files found in the dataset directory - please download the texture dataset')
 
     data = IQM_Vis.dataset_holder(files,
                                   metric,