From 14eea67f3b5dec592d4f2444ccb3f2dc171e17c0 Mon Sep 17 00:00:00 2001 From: Hussein Awala Date: Sat, 9 Sep 2023 21:56:38 +0200 Subject: [PATCH] Replace assert by if...raise in decorators package --- airflow/decorators/base.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/airflow/decorators/base.py b/airflow/decorators/base.py index 31a354e02018d..57362e73312a8 100644 --- a/airflow/decorators/base.py +++ b/airflow/decorators/base.py @@ -544,7 +544,8 @@ def __attrs_post_init__(self): def _expand_mapped_kwargs(self, context: Context, session: Session) -> tuple[Mapping[str, Any], set[int]]: # We only use op_kwargs_expand_input so this must always be empty. - assert self.expand_input is EXPAND_INPUT_EMPTY + if self.expand_input is not EXPAND_INPUT_EMPTY: + raise AssertionError(f"unexpected expand_input: {self.expand_input}") op_kwargs, resolved_oids = super()._expand_mapped_kwargs(context, session) return {"op_kwargs": op_kwargs}, resolved_oids