Strip trailing slashes from paths

Some file systems may add trailing slashing when listing folder contents
into-ai · Apr 3, 2020 · 8c877c7 · 8c877c7
1 parent 2c17d7b
commit 8c877c7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deeplearning2020/datasets.py b/deeplearning2020/datasets.py
@@ -62,7 +62,7 @@ def __init__(self, dataset: str, file_path: str = None) -> None:
         print(f"Loaded {self.image_count} images")
 
         self.raw_class_names = [
-            item for item in tf.io.gfile.listdir(data_dir) if item != "LICENSE.txt"
+            item.strip("/") for item in tf.io.gfile.listdir(data_dir) if item != "LICENSE.txt"
         ]
         self.raw_class_names.sort()