Replace assert by if...raise in decorators package (#34250)

(cherry picked from commit 0470f3a)
apache · Oct 5, 2023 · c554233 · c554233
1 parent 9f369ba
commit c554233
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/airflow/decorators/base.py b/airflow/decorators/base.py
@@ -523,7 +523,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