address comment

metadata-ingestion/src/datahub_provider/_plugin.py

@@ -1,6 +1,7 @@
 from datahub_provider._airflow_compat import Operator
 
 import contextlib
+import logging
 import traceback
 from typing import Any, Iterable, List
 
@@ -16,6 +17,7 @@
 from datahub_provider.hooks.datahub import DatahubGenericHook
 from datahub_provider.lineage.datahub import DatahubLineageConfig
 
+logger = logging.getLogger(__name__)
 
 def get_lineage_config() -> DatahubLineageConfig:
     """Load the lineage config from airflow.cfg."""
@@ -275,6 +277,7 @@ def custom_on_success_callback(context):
 
 
 def task_policy(task: BaseOperator) -> None:
+    task.log.debug(f"Setting task policy for Dag: {task.dag_id} Task: {task.task_id}")
     # task.add_inlets(["auto"])
     # task.pre_execute = _wrap_pre_execution(task.pre_execute)
     task.on_failure_callback = _wrap_on_failure_callback(task.on_failure_callback)
@@ -301,6 +304,8 @@ def _patch_policy(settings):
 
 
 def _patch_datahub_policy():
+    logger.info("Patching datahub policy")
+
     with contextlib.suppress(ImportError):
         import airflow_local_settings