Lift pandas pinning (#67)

Author: fjetter

.github/workflows/pre-commit.yml

@@ -9,4 +9,4 @@ jobs:
     steps:
       - uses: actions/checkout@v2
       - uses: actions/setup-python@v2
-      - uses: pre-commit/action@v2.0.0
+      - uses: pre-commit/action@v3.0.1

.github/workflows/tests.yml

@@ -26,7 +26,6 @@ jobs:
       - name: Setup Conda Environment
         uses: conda-incubator/setup-miniconda@v3
         with:
-          miniforge-variant: Mambaforge
           miniforge-version: latest
           use-mamba: true
           channel-priority: strict

.github/workflows/wheels.yml

@@ -23,16 +23,16 @@ jobs:
           fetch-depth: 0
       - uses: actions/setup-python@v4
         with:
-          python-version: '3.10' 
+          python-version: '3.10'
       - name: Build wheels
         run: pip wheel . -w dist
       - name: Build Source Dist
         run: python setup.py sdist
-      - uses: actions/upload-artifact@v3
+      - uses: actions/upload-artifact@v4
         with:
           name: wheel
           path: ./dist/dask_snowflake*
-      - uses: actions/upload-artifact@v3
+      - uses: actions/upload-artifact@v4
         with:
           name: sdist
           path: ./dist/dask-snowflake*
@@ -42,11 +42,11 @@ jobs:
     needs: [artifacts]
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/download-artifact@v3
+      - uses: actions/download-artifact@v4
         with:
           name: sdist
           path: dist
-      - uses: actions/download-artifact@v3
+      - uses: actions/download-artifact@v4
         with:
           name: wheel
           path: dist
@@ -65,11 +65,11 @@ jobs:
     permissions:
       id-token: write
     steps:
-      - uses: actions/download-artifact@v3
+      - uses: actions/download-artifact@v4
         with:
           name: sdist
           path: dist
-      - uses: actions/download-artifact@v3
+      - uses: actions/download-artifact@v4
         with:
           name: wheel
           path: dist

ci/environment-3.10.yaml

@@ -6,13 +6,9 @@ dependencies:
   - python=3.10
   - dask
   - distributed
-  # `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
+  - pandas
   - pyarrow
   - snowflake-connector-python >=2.6.0
   - snowflake-sqlalchemy
   # Testing
-  - pytest
+  - pytest

ci/environment-3.11.yaml

@@ -6,13 +6,9 @@ dependencies:
   - python=3.11
   - dask
   - distributed
-  # `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
+  - pandas
   - pyarrow
   - snowflake-connector-python >=2.6.0
   - snowflake-sqlalchemy
   # Testing
-  - pytest
+  - pytest

ci/environment-3.12.yaml

@@ -6,13 +6,9 @@ dependencies:
   - python=3.12
   - dask
   - distributed
-  # `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
+  - pandas
   - pyarrow
   - snowflake-connector-python >=2.6.0
   - snowflake-sqlalchemy
   # Testing
-  - pytest
+  - pytest

ci/environment-3.9.yaml

@@ -6,13 +6,9 @@ dependencies:
   - python=3.9
   - dask
   - distributed
-  # `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
+  - pandas
   - pyarrow
   - snowflake-connector-python >=2.6.0
   - snowflake-sqlalchemy
   # Testing
-  - pytest
+  - pytest

dask_snowflake/tests/test_core.py

@@ -71,7 +71,7 @@ def test_read_empty_result(table, connection_kwargs, client):
 
     result = read_snowflake(
         f"SELECT * FROM {table} where A > %(target)s",
-        execute_params={"target": df.A.max()},
+        execute_params={"target": df.A.max().item()},
         connection_kwargs=connection_kwargs,
         npartitions=2,
     )