safely check warnings

datahub-project · Nov 10, 2022 · c969105 · c969105
1 parent ae4825d
commit c969105
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/metadata-ingestion/src/datahub/ingestion/source/tableau.py b/metadata-ingestion/src/datahub/ingestion/source/tableau.py
@@ -383,7 +383,10 @@ def get_connection_object_page(
 
         if "errors" in query_data:
             errors = query_data["errors"]
-            if all(error["extensions"]["severity"] == "WARNING" for error in errors):
+            if all(
+                error.get("extensions", {}).get("severity", None) == "WARNING"
+                for error in errors
+            ):
                 self.report.report_warning(key=connection_type, reason=f"{errors}")
             else:
                 raise RuntimeError(f"Query {connection_type} error: {errors}")