Fix mypy errors in apache/drill/operators and /apache/pig/operators (#…

…20597)
apache · Dec 30, 2021 · a1d457a · a1d457a
1 parent 0bf424f
commit a1d457a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/airflow/providers/apache/drill/operators/drill.py b/airflow/providers/apache/drill/operators/drill.py
@@ -62,7 +62,7 @@ def __init__(
         self.sql = sql
         self.drill_conn_id = drill_conn_id
         self.parameters = parameters
-        self.hook = None
+        self.hook: Optional[DrillHook] = None
 
     def execute(self, context: 'Context'):
         self.log.info('Executing: %s on %s', self.sql, self.drill_conn_id)

diff --git a/airflow/providers/apache/pig/operators/pig.py b/airflow/providers/apache/pig/operators/pig.py
@@ -65,7 +65,7 @@ def __init__(
         self.pig = pig
         self.pig_cli_conn_id = pig_cli_conn_id
         self.pig_opts = pig_opts
-        self.hook = None
+        self.hook: Optional[PigCliHook] = None
 
     def prepare_template(self):
         if self.pigparams_jinja_translate: