From f5769cb7cbaa25c43524239c4d71de584f3ce94e Mon Sep 17 00:00:00 2001 From: Phillip Cloud <417981+cpcloud@users.noreply.github.com> Date: Wed, 24 Jul 2024 18:17:37 -0400 Subject: [PATCH] test: use `from_connection` in snowflake udf test --- ibis/backends/snowflake/tests/conftest.py | 2 +- ibis/backends/snowflake/tests/test_udf.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/ibis/backends/snowflake/tests/conftest.py b/ibis/backends/snowflake/tests/conftest.py index 4aeb3d8e0dea..8b309d71d7db 100644 --- a/ibis/backends/snowflake/tests/conftest.py +++ b/ibis/backends/snowflake/tests/conftest.py @@ -209,7 +209,7 @@ def connect(*, tmpdir, worker_id, **kw) -> BaseBackend: "schema": os.environ["SNOWFLAKE_SCHEMA"], } ) - return ibis.backends.snowflake.Backend.from_snowpark(builder.create()) + return ibis.backends.snowflake.Backend.from_connection(builder.create()) else: return ibis.connect(_get_url(), **kw) diff --git a/ibis/backends/snowflake/tests/test_udf.py b/ibis/backends/snowflake/tests/test_udf.py index 4a59013cebec..f73f79c17380 100644 --- a/ibis/backends/snowflake/tests/test_udf.py +++ b/ibis/backends/snowflake/tests/test_udf.py @@ -238,7 +238,7 @@ def test_ibis_inside_snowpark(snowpark_session, execute_as): def ibis_sproc(session): import ibis.backends.snowflake - con = ibis.backends.snowflake.Backend.from_snowpark(session) + con = ibis.backends.snowflake.Backend.from_connection(session) expr = ( con.tables.functional_alltypes.group_by("string_col")