From bf2f6461d25e7864b0222d7e2527c320194ab804 Mon Sep 17 00:00:00 2001
From: tariqshihadah <tariq.shihadah@gmail.com>
Date: Tue, 20 Aug 2024 15:34:59 -0500
Subject: [PATCH] enforce dtypes on events union call

---
 linref/events/union.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/linref/events/union.py b/linref/events/union.py
index 7af06cb..038f943 100644
--- 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: