Обновить test_clickhouse_dag.py

test_clickhouse_dag.py

@@ -1,11 +1,15 @@
 from airflow import DAG
-from airflow.providers.clickhouse.hooks.clickhouse import ClickHouseHook
 from airflow.operators.python import PythonOperator
 from datetime import datetime
 
+# Импортируем hook из установленного пакета airflow-providers-clickhouse
+from airflow_clickhouse_plugin.hooks.clickhouse import ClickHouseHook
+
 def test_conn():
-    hook = ClickHouseHook(clickhouse_conn_id="my_clickhouse_provider")
-    hook.run("SELECT 1")
+    hook = ClickHouseHook(
+        clickhouse_conn_id="my_clickhouse_provider"  # ваш Connection ID в Airflow
+    )
+    hook.execute("SELECT 1")  # у этого пакета метод называется execute
 
 with DAG(
     dag_id="test_clickhouse_dag",
@@ -13,4 +17,7 @@ with DAG(
     schedule=None,
     catchup=False,
 ) as dag:
-    PythonOperator(task_id="check_clickhouse", python_callable=test_conn)
+    PythonOperator(
+        task_id="check_clickhouse",
+        python_callable=test_conn,
+    )