Update local.py

astronomer · Jan 17, 2025 · ff8deaf · ff8deaf
1 parent 8f4d2bd
commit ff8deaf
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/cosmos/operators/local.py b/cosmos/operators/local.py
@@ -47,10 +47,10 @@
     from airflow.datasets import Dataset  # noqa: F811
     from dbt.cli.main import dbtRunner, dbtRunnerResult
 
-    try:
-        from openlineage.client.event_v2 import RunEvent
-    except ImportError:
-        from openlineage.client.run import RunEvent
+    try: # pragma: no cover 
+        from openlineage.client.event_v2 import RunEvent # pragma: no cover 
+    except ImportError: # pragma: no cover 
+        from openlineage.client.run import RunEvent # pragma: no cover 
 
 
 from sqlalchemy.orm import Session