fix int32 instead of int8

rapidsai · Aug 8, 2024 · 6279647 · 6279647
1 parent 5f9be85
commit 6279647
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/cuspatial/cuspatial/core/_column/geometa.py b/python/cuspatial/cuspatial/core/_column/geometa.py
@@ -49,11 +49,11 @@ def __init__(
             meta_uo = meta["union_offsets"]
             if isinstance(meta_uo, cudf.core.column.ColumnBase):
                 self.union_offsets = cudf.Series._from_column(meta_uo).astype(
-                    "int8"
+                    "int32"
                 )
             else:
                 # Could be Series from GeoSeries.__getitem__
-                self.union_offsets = cudf.Series(meta_uo, dtype="int8")
+                self.union_offsets = cudf.Series(meta_uo, dtype="int32")
         else:
             self.input_types = cudf.Series(meta.input_types, dtype="int8")
             self.union_offsets = cudf.Series(meta.union_offsets, dtype="int32")