Merge pull request fastai#204 from Gokkulnath/patch-2

Avoid "NotADirectoryError" on .DS_Store fastai#200
dineshkumarsarangapani · Mar 10, 2018 · ca88790 · ca88790
2 parents 3ff8bed + aa0150e
commit ca88790
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fastai/dataset.py b/fastai/dataset.py
@@ -43,7 +43,7 @@ def read_dirs(path, folder):
     labels, filenames, all_labels = [], [], []
     full_path = os.path.join(path, folder)
     for label in sorted(os.listdir(full_path)):
-        if label not in ('.ipynb_checkpoints'):
+        if label not in ('.ipynb_checkpoints','.DS_Store'):
             all_labels.append(label)
             for fname in os.listdir(os.path.join(full_path, label)):
                 filenames.append(os.path.join(folder, label, fname))