Handle the errors properly

NPodlozhniy · Jan 16, 2023 · 72af675 · 72af675
1 parent 076ca10
commit 72af675
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/dataloader.py b/dataloader.py
@@ -73,8 +73,10 @@ def get_reality(wks_name, *kwargs) -> list:
     for wks_name in wks_names(button):
         try:
             ref_dict[wks_name] = ref_dict[wks_name].result()
-        except Exception as e:
-            print(e, flush=True)
+        except RuntimeError: # if there is no tab
+            ref_dict.pop(wks_name)
+        except IndexError: # if there is no data
+            ref_dict.pop(wks_name)
         finally:
             fact_dict[wks_name] = fact_dict[wks_name].result()