From 2acdb55fd1076ebecd85adb09f5781e255a9c265 Mon Sep 17 00:00:00 2001 From: Valay Dave Date: Mon, 30 Jan 2023 19:46:20 +0000 Subject: [PATCH 1/2] exception when Airflow with timezones. --- metaflow/plugins/airflow/airflow_cli.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/metaflow/plugins/airflow/airflow_cli.py b/metaflow/plugins/airflow/airflow_cli.py index e9ea4521e28..3245e09bfb4 100644 --- a/metaflow/plugins/airflow/airflow_cli.py +++ b/metaflow/plugins/airflow/airflow_cli.py @@ -400,6 +400,15 @@ def _validate_workflow(flow, graph, flow_datastore, metadata, workflow_timeout): ) ) + schedule = flow._flow_decorators.get("schedule") + if not schedule: + return + + if schedule.timezone is not None: + raise AirflowException( + "`airflow create` does not support scheduling with `timezone`." + ) + def resolve_dag_name(name): project = current.get("project_name") From 8586fe1039ff462fc6cb82d04b124addc2c21f7b Mon Sep 17 00:00:00 2001 From: Valay Dave Date: Mon, 30 Jan 2023 19:55:14 +0000 Subject: [PATCH 2/2] Fix bug with multi decorator --- metaflow/plugins/airflow/airflow_cli.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/metaflow/plugins/airflow/airflow_cli.py b/metaflow/plugins/airflow/airflow_cli.py index 3245e09bfb4..c938f5e450c 100644 --- a/metaflow/plugins/airflow/airflow_cli.py +++ b/metaflow/plugins/airflow/airflow_cli.py @@ -403,7 +403,8 @@ def _validate_workflow(flow, graph, flow_datastore, metadata, workflow_timeout): schedule = flow._flow_decorators.get("schedule") if not schedule: return - + + schedule = schedule[0] if schedule.timezone is not None: raise AirflowException( "`airflow create` does not support scheduling with `timezone`."