fix case render

mindsdb · Nov 10, 2024 · 4298b66 · 4298b66
1 parent 65b3d05
commit 4298b66
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/mindsdb_sql/render/sqlalchemy_render.py b/mindsdb_sql/render/sqlalchemy_render.py
@@ -297,7 +297,11 @@ def prepare_case(self, t: ast.Case):
         if t.default is not None:
             default = self.to_expression(t.default)
 
-        return sa.case(*conditions, else_=default)
+        value = None
+        if t.arg is not None:
+            value = self.to_expression(t.arg)
+
+        return sa.case(*conditions, else_=default, value=value)
 
     def to_function(self, t):
         op = getattr(sa.func, t.op)