Fix mypy spark hooks (#20290)

apache · Dec 14, 2021 · 1a2a249 · 1a2a249
1 parent b20e6d3
commit 1a2a249
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/airflow/providers/apache/spark/hooks/spark_sql.py b/airflow/providers/apache/spark/hooks/spark_sql.py
@@ -17,7 +17,7 @@
 # under the License.
 #
 import subprocess
-from typing import TYPE_CHECKING, Any, List, Optional, Union
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union
 
 from airflow.exceptions import AirflowException, AirflowNotFoundException
 from airflow.hooks.base import BaseHook
@@ -88,9 +88,10 @@ def __init__(
             conn: "Optional[Connection]" = self.get_connection(conn_id)
         except AirflowNotFoundException:
             conn = None
-            options = {}
+            options: Dict = {}
         else:
-            options = conn.extra_dejson
+            if conn:
+                options = conn.extra_dejson
 
         # Set arguments to values set in Connection if not explicitly provided.
         if master is None: