Skip to content

Commit

Permalink
Fix MyPy Errors for Presto provider (#20244)
Browse files Browse the repository at this point in the history
  • Loading branch information
kazanzhy committed Dec 13, 2021
1 parent 65d7f42 commit 9c05a95
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions airflow/providers/presto/hooks/presto.py
Expand Up @@ -16,7 +16,7 @@
# specific language governing permissions and limitations
# under the License.
import os
from typing import Any, Iterable, Optional
from typing import Any, Callable, Iterable, Optional

import prestodb
from prestodb.exceptions import DatabaseError
Expand Down Expand Up @@ -148,9 +148,10 @@ def run(
hql,
autocommit: bool = False,
parameters: Optional[dict] = None,
handler: Optional[Callable] = None,
) -> None:
"""Execute the statement against Presto. Can be used to create views."""
return super().run(sql=self._strip_sql(hql), parameters=parameters)
return super().run(sql=self._strip_sql(hql), parameters=parameters, handler=handler)

def insert_rows(
self,
Expand Down

0 comments on commit 9c05a95

Please sign in to comment.