fix: prevent KeyError for getting inspect data

biothings · Apr 10, 2023 · 039b846 · 039b846
1 parent 8275843
commit 039b846
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/biothings/hub/datainspect/inspector.py b/biothings/hub/datainspect/inspector.py
@@ -331,7 +331,7 @@ def flatten(self, data_provider, mode=("type", "stats"), do_validate=True):
 
         results = {}
         if data_provider_type == "source":
-            src_sources_inspect_data = registerer_obj.src_doc["inspect"]["jobs"]
+            src_sources_inspect_data = registerer_obj.src_doc.get("inspect", {}).get("jobs", {})
             for src_source_name, inspect_data in src_sources_inspect_data.items():
                 results[src_source_name] = {
                     _mode: btinspect.flatten_and_validate(
@@ -340,7 +340,7 @@ def flatten(self, data_provider, mode=("type", "stats"), do_validate=True):
                     for _mode in mode
                 }
         else:
-            inspect_data = registerer_obj.src_build["inspect"]["results"]
+            inspect_data = registerer_obj.src_build.get("inspect", {}).get("results", {})
             results[backend_provider] = {
                 _mode: btinspect.flatten_and_validate(inspect_data.get(_mode) or {}, do_validate) for _mode in mode
             }