From f73f69bf41a79fd84b05c93a0e2a98d7cd3950c2 Mon Sep 17 00:00:00 2001
From: Mike Boyle <michael.oliver.boyle@gmail.com>
Date: Thu, 24 Oct 2024 10:05:18 -0400
Subject: [PATCH] Ensure that IMR runs are only BBH [skip ci]

---
 sxs/simulations/simulations.py | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/sxs/simulations/simulations.py b/sxs/simulations/simulations.py
index c80eff3..05034a0 100644
--- a/sxs/simulations/simulations.py
+++ b/sxs/simulations/simulations.py
@@ -6,10 +6,10 @@
 
 class SimulationsDataFrame(pd.DataFrame):
     @property
-    def BHBH(self):
+    def BBH(self):
         """Restrict dataframe to just binary black hole systems"""
         return type(self)(self[self["object_types"] == "BHBH"])
-    BBH = BHBH
+    BHBH = BBH
     
     @property
     def BHNS(self):
@@ -65,9 +65,10 @@ def IMR(self):
         measured eccentricity means that the system is not
         hyperbolic or head-on.
         """
-        return type(self)(self[
-            np.isfinite(self["reference_eccentricity"])
-            & np.isfinite(self["remnant_mass"])
+        df = self.BBH
+        return type(df)(df[
+            np.isfinite(df["reference_eccentricity"])
+            & np.isfinite(df["remnant_mass"])
         ])