BUG: fix index in spatial partitions (#329)

Author: martinfleis

.github/workflows/tests.yaml

@@ -24,7 +24,6 @@ jobs:
     defaults:
       run:
         shell: bash -l {0}
-    continue-on-error: true
     strategy:
       matrix:
         os: [ubuntu-latest]

CHANGELOG.md

@@ -6,7 +6,11 @@ Version 0.5.0 (upcoming)
 
 Deprecations and compatibility notes:
 
-- The deprecated `geom_almost_equals` method has been removed. Use `geom_equals_exact` instead.
+- The deprecated `geom_almost_equals` method has been removed. Use `geom_equals_exact` instead (#326).
+
+Bug fixes:
+
+- Fixed index in spatial partitions (#329).
 
 
 Version 0.4.3 (January, 2025)

dask_geopandas/expr.py

@@ -233,7 +233,7 @@ def calculate_spatial_partitions(self):
                 )
             ).compute(),
             crs=self.crs,
-        )
+        ).reset_index(drop=True)
         self.spatial_partitions = parts
 
     def _propagate_spatial_partitions(self, new_object):

dask_geopandas/tests/io/test_arrow.py

@@ -92,7 +92,10 @@ def test_roundtrip(tmp_path, naturalearth_lowres):
     assert_geodataframe_equal(result_gpd, df)
     # reading back also populates the spatial partitioning property
     ddf.calculate_spatial_partitions()
-    assert_geoseries_equal(result.spatial_partitions, ddf.spatial_partitions.envelope)
+    assert_geoseries_equal(
+        result.spatial_partitions,
+        ddf.spatial_partitions.envelope,
+    )
 
 
 def test_roundtrip_s3(s3_resource, s3_storage_options, naturalearth_lowres):