Merge pull request MrGreyfun#22 from MrGreyfun/greyfun

Bug fix: a error is raised when refresh file display if there is a fold in the work_dir
zh3305 · Nov 3, 2023 · 0526391 · 0526391
2 parents 62370dd + c869259
commit 0526391
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/web_ui.py b/src/web_ui.py
@@ -67,9 +67,9 @@ def refresh_file_display(state_dict: Dict) -> List[str]:
     filenames = os.listdir(work_dir)
     paths = []
     for filename in filenames:
-        paths.append(
-            os.path.join(work_dir, filename)
-        )
+        path = os.path.join(work_dir, filename)
+        if not os.path.isdir(path):
+            paths.append(path)
     return paths