enforce dtypes on events union call

tariqshihadah · Aug 20, 2024 · bf2f646 · bf2f646
1 parent 6e754a3
commit bf2f646
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/linref/events/union.py b/linref/events/union.py
@@ -209,6 +209,11 @@ def union(
                 **{col: arr for col, arr in zip(self.objs[0].keys, keys.T)},
                 **{self.objs[0].beg: begs, self.objs[0].end: ends},
             })
+        # Enforce dtypes
+        dtypes = {col: self.objs[0].df.dtypes[col] for col in self.objs[0].keys}
+        dtypes[self.objs[0].beg] = float
+        dtypes[self.objs[0].end] = float
+        data = data.astype(dtypes, copy=False, errors='ignore')
 
         # Merge resegmented data with original dataframe columns
         if merge and get_index: