From 7dab06519dc68d56b62518c16d34130ad6df79f5 Mon Sep 17 00:00:00 2001
From: Swaroop Jagadish <67564030+swaroopjagadish@users.noreply.github.com>
Date: Mon, 11 Oct 2021 20:58:37 -0700
Subject: [PATCH] fix(ci): fix task instance constructor in airflow tests
 (#3366)

---
 metadata-ingestion/tests/unit/test_airflow.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/metadata-ingestion/tests/unit/test_airflow.py b/metadata-ingestion/tests/unit/test_airflow.py
index ef25f1844d..4b310f4216 100644
--- a/metadata-ingestion/tests/unit/test_airflow.py
+++ b/metadata-ingestion/tests/unit/test_airflow.py
@@ -233,7 +233,11 @@ def test_lineage_backend(mock_emit, inlets, outlets):
             )
             op1 >> op2
 
-        ti = TI(task=op2, execution_date=DEFAULT_DATE)
+        if airflow.version.version.startswith("1"):
+            ti = TI(task=op2, execution_date=DEFAULT_DATE)
+        else:
+            ti = TI(task=op2)
+
         ctx1 = {
             "dag": dag,
             "task": op2,