Remove restriction on pandas version (#64)

coiled · Nov 14, 2024 · 3ae24f7 · 3ae24f7
1 parent 7ddddcd
commit 3ae24f7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git 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
@@ -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