Change log levels to analyse behaviour in Astro CLI

astronomer · Dec 20, 2024 · aaa19da · aaa19da
1 parent 8919ef0
commit aaa19da
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/cosmos/listeners/dag_run_listener.py b/cosmos/listeners/dag_run_listener.py
@@ -1,7 +1,5 @@
 from __future__ import annotations
 
-import functools
-
 from airflow.listeners import hookimpl
 from airflow.models.dag import DAG
 from airflow.models.dagrun import DagRun
@@ -27,7 +25,6 @@ def is_cosmos_dag(dag: DAG) -> bool:
 
 
 def total_cosmos_task_groups(dag: DAG) -> int:
-
     cosmos_task_groups = 0
     for group_id, task_group in dag.task_group_dict.items():
         if task_group.__class__.__module__.startswith("cosmos."):
@@ -45,7 +42,6 @@ def total_cosmos_tasks(dag: DAG) -> int:
     return cosmos_tasks
 
 
-@functools.lru_cache()
 def uses_cosmos(dag: DAG) -> bool:
     return bool(is_cosmos_dag(dag) or total_cosmos_task_groups(dag) or total_cosmos_tasks(dag))
 

diff --git a/cosmos/telemetry.py b/cosmos/telemetry.py
@@ -44,7 +44,7 @@ def emit_usage_metrics(metrics: dict[str, object]) -> bool:
     telemetry_url = constants.TELEMETRY_URL.format(
         **metrics, telemetry_version=constants.TELEMETRY_VERSION, query_string=query_string
     )
-    logging.debug("Telemetry is enabled. Emitting the following usage metrics to %s: %s", telemetry_url, metrics)
+    logging.info("Telemetry is enabled. Emitting the following usage metrics to %s: %s", telemetry_url, metrics)
     response = httpx.get(telemetry_url, timeout=constants.TELEMETRY_TIMEOUT, follow_redirects=True)
     if not response.is_success:
         logging.warning(
@@ -71,5 +71,5 @@ def emit_usage_metrics_if_enabled(event_type: str, additional_metrics: dict[str,
         is_success = emit_usage_metrics(metrics)
         return is_success
     else:
-        logging.debug("Telemetry is disabled. To enable it, export AIRFLOW__COSMOS__ENABLE_TELEMETRY=True.")
+        logging.info("Telemetry is disabled. To enable it, export AIRFLOW__COSMOS__ENABLE_TELEMETRY=True.")
         return False