try except

nerdai · Sep 24, 2024 · 58175e8 · 58175e8
1 parent 0156d5a
commit 58175e8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/arc_finetuning_st/streamlit/controller.py b/arc_finetuning_st/streamlit/controller.py
@@ -157,7 +157,10 @@ def saved_finetuning_examples(self) -> List[str]:
 
     @property
     def task_file_names(self) -> List[str]:
-        return listdir(self._data_path)
+        try:
+            return listdir(self._data_path)
+        except FileNotFoundError:
+            return []
 
     def radio_format_task_name(self, selected_task: str) -> str:
         if selected_task in self.saved_finetuning_examples: