Update clickhouse.py

bryzgaloff · May 30, 2024 · f40ec4f · f40ec4f
1 parent 85a60cf
commit f40ec4f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/airflow_clickhouse_plugin/hooks/clickhouse.py b/src/airflow_clickhouse_plugin/hooks/clickhouse.py
@@ -91,7 +91,7 @@ def execute(
 
     def get_pandas_df(
         self, sql: str, parameters: t.Union[None, dict, list, tuple, t.Generator] = None
-    ) -> Any:
+    ) -> t.Any:
         import pandas as pd
 
         rows, columns_defs = self.run(sql, parameters, with_column_types=True)
@@ -104,7 +104,7 @@ def run(
         parameters: t.Union[None, dict, list, tuple, t.Generator] = None,
         with_column_types: bool = False,
         types_check: bool = False,
-    ) -> Any:
+    ) -> t.Any:
         if isinstance(sql, str):
             sql = (sql,)
         with _disconnecting(self.get_conn()) as conn: