diff --git a/plugins/flytekit-airflow/flytekitplugins/airflow/task.py b/plugins/flytekit-airflow/flytekitplugins/airflow/task.py index 04fb848cb5..225115d12d 100644 --- 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 index 91214e6dbf..682cd72c18 100644 --- 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"