place holder

snowflakedb · Feb 6, 2025 · 08643fc · 08643fc
1 parent 24c6a2a
commit 08643fc
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/src/snowflake/snowpark/_internal/proto/ast.proto b/src/snowflake/snowpark/_internal/proto/ast.proto
@@ -4,6 +4,7 @@
 syntax = "proto3";
 
 option java_package = "com.snowflake.snowpark.proto";
+option java_outer_classname = "JavaProto";
 
 package ast;
 

diff --git a/src/snowflake/snowpark/functions.py b/src/snowflake/snowpark/functions.py
@@ -717,7 +717,7 @@ def bitshiftright_unsigned(
         bitshiftright(c, n, _emit_ast=False),
         _emit_ast=False,
     )
-    col = call_builtin("bitand", unsigned_c, max_bit - 1, _emit_ast=_emit_ast)
+    col = call_builtin("bitand", unsigned_c, max_bit - 1, _emit_ast=False)
     col._ast = ast
     return col
 

diff --git a/tests/ast/data/functions2.test b/tests/ast/data/functions2.test
@@ -644,6 +644,8 @@ df314 = df.select(instr("A", "test_str"))
 
 df315 = df.select(nth_value("A", 2, False), nth_value("A", 2, True), nth_value(col("B"), 2, False))
 
+df316 = df.select(bitshiftright_unsigned("A", 2), bitshiftright_unsigned("A", col("B")))
+
 ## EXPECTED ENCODED AST
 
 interned_value_table {