add _emit_ast to CaseExpr

snowflakedb · Oct 24, 2024 · 1d631fc · 1d631fc
1 parent 4336b30
commit 1d631fc
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/snowflake/snowpark/column.py b/src/snowflake/snowpark/column.py
@@ -1417,11 +1417,9 @@ class CaseExpr(Column):
     """
 
     def __init__(
-        self,
-        expr: CaseWhen,
-        _ast: Optional[proto.Expr] = None,
+        self, expr: CaseWhen, _ast: Optional[proto.Expr] = None, _emit_ast: bool = True
     ) -> None:
-        super().__init__(expr, _ast=_ast)
+        super().__init__(expr, _ast=_ast, _emit_ast=_emit_ast)
         self._branches = expr.branches
 
     @publicapi
@@ -1453,6 +1451,7 @@ def when(
                 ]
             ),
             _ast=self._ast,
+            _emit_ast=_emit_ast,
         )
 
     @publicapi