Skip to content

Commit

Permalink
Ensure that IMR runs are only BBH [skip ci]
Browse files Browse the repository at this point in the history
  • Loading branch information
moble committed Oct 24, 2024
1 parent 3ac4431 commit f73f69b
Showing 1 changed file with 6 additions and 5 deletions.
11 changes: 6 additions & 5 deletions sxs/simulations/simulations.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down Expand Up @@ -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"])
])


Expand Down

0 comments on commit f73f69b

Please sign in to comment.