Add back index

rapidsai · Aug 6, 2024 · 75eaf93 · 75eaf93
1 parent 34e9097
commit 75eaf93
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/cuspatial/cuspatial/core/geodataframe.py b/python/cuspatial/cuspatial/core/geodataframe.py
@@ -222,6 +222,8 @@ def _gather(self, gather_map: GatherMap, keep_index=True):
             for geo in geo_data.keys()
         }
         geo_gathered = GeoDataFrame(gathered)
+        # _gather may have discarded the index, so re-add
+        cudf_gathered.index = geo_gathered.index
 
         # combine
         return GeoDataFrame._from_data(