From c5542335c6bac2fdc94ed973e365b522e56aaaf5 Mon Sep 17 00:00:00 2001
From: Hussein Awala <hussein@awala.fr>
Date: Sat, 9 Sep 2023 23:13:22 +0200
Subject: [PATCH] Replace assert by if...raise in decorators package (#34250)

(cherry picked from commit 0470f3af6ddeda0917cd3a0616fa3206bad00236)
---
 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 2594ec23995f4..1a4bc06253895 100644
--- 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