Wrap in text

coiled · Nov 13, 2024 · 285cfeb · 285cfeb
1 parent 620f3c5
commit 285cfeb
Showing 1 changed file with 2 additions and 2 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
@@ -30,7 +30,7 @@ def table(connection_kwargs):
 
     engine = create_engine(URL(**connection_kwargs))
     with engine.connect() as conn:
-        conn.execute(f"DROP TABLE IF EXISTS {name}")
+        conn.execute(text(f"DROP TABLE IF EXISTS {name}"))
 
 
 @pytest.fixture(scope="module")