Fixed 'NotADirectoryError' caused by .ds_store file under macos (#86)

Co-authored-by: lizg <[email protected]>
opendatalab · Jan 16, 2025 · 7c4be36 · 7c4be36
1 parent 1724abc
commit 7c4be36
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mesh-candidate_bestfit/map_dict.py b/mesh-candidate_bestfit/map_dict.py
@@ -14,6 +14,8 @@ def get_map_dict(use_aug):
     instance2pathlist = {}
     root_dir = './element_pool'
     for category in os.listdir(root_dir):
+        if category == '.DS_Store':
+            continue
         category_dir = os.path.join(root_dir,category)
         filelist = os.listdir(category_dir)
         for filename in tqdm(filelist):