BUG: fix index in spatial partitions

martinfleis · martinfleis · commit f8f98ea88a9c · 2025-06-02T20:59:19.000+02:00
diff --git a/dask_geopandas/expr.py b/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):
diff --git a/dask_geopandas/tests/io/test_arrow.py b/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):

Original file line number	Diff line number	Diff line change
`@@ -233,7 +233,7 @@ def calculate_spatial_partitions(self):`
`233`	`233`	`)`
`234`	`234`	`).compute(),`
`235`	`235`	`crs=self.crs,`
`236`		`- )`
	`236`	`+ ).reset_index(drop=True)`
`237`	`237`	`self.spatial_partitions = parts`
`238`	`238`
`239`	`239`	`def _propagate_spatial_partitions(self, new_object):`