在 Python 存储过程中使用过滤器时出现 _udf_code 错误

Question

一直在尝试开始在 Snowflake 中使用 Python 程序。我有另一个可以正常工作的基本程序，但我无法让这部分工作。我希望过滤一个数据框，但得到这个奇怪的错误。

[P0000][100357] Python Interpreter Error: Traceback (most recent call last): File "_udf_code.py", line 6, in run File "/usr/lib/python_udf/de--0d/lib/python3.8/site-packages/snowflake/snowpark/_internal/telemetry.py", line 133, in wrap result = func(*args, **kwar ...

这是存储过程，它相当简单

CREATE OR REPLACE PROCEDURE utility.procedure.RECREATE_STALE_STREAM_PYTHON()
  RETURNS STRING
  LANGUAGE PYTHON
  RUNTIME_VERSION = '3.8'
  PACKAGES = ('snowflake-snowpark-python')
  HANDLER = 'run'
AS
$$
from snowflake.snowpark.functions import col
def run(session):
    show_streams = "show streams in account;"
    streams = session.sql(show_streams)
    stale_streams = streams.filter(col('stale') == 'true').collect();
    return stale_streams
$$;

提前致谢

我已经尝试了我能想到的一切，似乎没有任何效果

Answer 1

您的 UDF 定义签名：

CREATE OR REPLACE PROCEDURE utility.procedure.RECREATE_STALE_STREAM_PYTHON()

与指定为处理程序的 Python 函数的签名不匹配：

def run(session):

UDF 定义在签名中具有零参数。处理程序的签名中有一个参数。它们需要在数量和类型上匹配（Python 中的松散类型是可以的，但必须能够被强制转换）。

这里有一个 Python UDF 示例，它显示了匹配的 UDF 定义签名和处理程序签名：

https://docs.snowflake.com/en/developer-guide/udf/python/udf-python-creating.html#specifying-multiple-import-files

UDF 签名：

create or replace function multiple_import_files(s string)

匹配处理程序的签名：

def compute(s):

一种选择是在 UDF 签名中传递 session 的值。但是，请注意名称，因为 session 是保留的 SQL 关键字。