Enhance Airflow Plugin (#2099)

Signed-off-by: Future Outlier <[email protected]> Co-authored-by: Future Outlier <[email protected]>
flyteorg · Jan 13, 2024 · 76f6db2 · 76f6db2
1 parent cc94d33
commit 76f6db2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/plugins/flytekit-airflow/flytekitplugins/airflow/task.py b/plugins/flytekit-airflow/flytekitplugins/airflow/task.py
@@ -160,7 +160,7 @@ def _get_airflow_instance(
     return obj_def(**airflow_obj.parameters)
 
 
-def _is_deferrable(cls: Type):
+def _is_deferrable(cls: Type) -> bool:
     """
     This function is used to check if the Airflow operator is deferrable.
     """

diff --git a/plugins/flytekit-airflow/setup.py b/plugins/flytekit-airflow/setup.py
@@ -7,6 +7,7 @@
 plugin_requires = [
     "apache-airflow",
     "flytekit>=1.9.0",
+    "flyteidl>=1.10.6",
 ]
 
 __version__ = "0.0.0+develop"