moar changes

snowflakedb · Jan 28, 2025 · c1ee08b · c1ee08b
1 parent a726012
commit c1ee08b
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/ast/decoder.py b/tests/ast/decoder.py
@@ -2821,7 +2821,7 @@ def decode_expr(self, expr: proto.Expr, **kwargs) -> Any:
                     name = self.decode_name_expr(expr.stored_procedure.name)
 
                 ret_sproc = sproc(
-                    self.session.sproc._registry[registered_object_name],
+                    self.session.sproc._registry[registered_object_name].func,
                     name=name,
                     return_type=return_type,
                     input_types=input_types,
@@ -2850,7 +2850,6 @@ def decode_expr(self, expr: proto.Expr, **kwargs) -> Any:
                         mode = "OBJECT"
 
                 if len(path) == 0:
-
                     return self.session.flatten(
                         input=input, outer=outer, recursive=recursive, mode=mode
                     )