add type coercion as extra safeguard

wildlife-dynamics · Nov 20, 2024 · 5d167a7 · 5d167a7
1 parent 288ed3d
commit 5d167a7
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/ecoscope/base/base.py b/ecoscope/base/base.py
@@ -104,6 +104,13 @@ def remove_filtered(self, inplace=False):
         else:
             frame = self.copy()
 
+        if not frame["junk_status"].dtype == bool:
+            warnings.warn(
+                f"junk_status column is of type {frame['junk_status'].dtype}, expected `bool`. "
+                "Attempting to automatically convert."
+            )
+            frame["junk_status"] = frame["junk_status"].astype(bool)
+
         frame.query("~junk_status", inplace=True)
 
         if not inplace: