diff --git a/dask_snowflake/tests/test_core.py b/dask_snowflake/tests/test_core.py index 70a054c..58e8b5f 100644 --- a/dask_snowflake/tests/test_core.py +++ b/dask_snowflake/tests/test_core.py @@ -5,7 +5,7 @@ import pytest import snowflake.connector from snowflake.sqlalchemy import URL -from sqlalchemy import create_engine +from sqlalchemy import create_engine, text import dask import dask.dataframe as dd @@ -29,7 +29,8 @@ def table(connection_kwargs): yield name engine = create_engine(URL(**connection_kwargs)) - engine.execute(f"DROP TABLE IF EXISTS {name}") + with engine.connect() as conn: + conn.execute(text(f"DROP TABLE IF EXISTS {name}")) @pytest.fixture(scope="module") diff --git a/requirements.txt b/requirements.txt index 1c4a6ee..85165a6 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,8 +2,3 @@ dask>=2024.3.0 distributed snowflake-connector-python[pandas]>=2.6.0 snowflake-sqlalchemy -# `pandas=2.2` dropped support for `sqlalchemy<2`, but `snowflake-sqlalchemy` -# doesn't support `sqlalchemy>=2` yet. Temporarily pinning `pandas<2.2` for now. -# xref https://github.com/pandas-dev/pandas/issues/57049 -# xref https://github.com/snowflakedb/snowflake-sqlalchemy/issues/380 -pandas<2.2