From 002843eb90acfdf7eba15607109d2ad18ac288fb Mon Sep 17 00:00:00 2001 From: Kaze Wong Date: Sun, 26 May 2024 15:15:07 -0400 Subject: [PATCH] Update waveform.py --- src/jimgw/single_event/waveform.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/jimgw/single_event/waveform.py b/src/jimgw/single_event/waveform.py index b763e153..e2084ea1 100644 --- a/src/jimgw/single_event/waveform.py +++ b/src/jimgw/single_event/waveform.py @@ -93,7 +93,8 @@ def __init__(self, f_ref: float = 20.0, use_lambda_tildes: bool = False): self.f_ref = f_ref self.use_lambda_tildes = use_lambda_tildes - def __call__(self, frequency: Float[Array, " n_dim"], params: dict[str, Float] + def __call__( + self, frequency: Float[Array, " n_dim"], params: dict[str, Float] ) -> dict[str, Float[Array, " n_dim"]]: output = {} @@ -152,7 +153,8 @@ def __init__( self.use_lambda_tildes = use_lambda_tildes self.no_taper = no_taper - def __call__(self, frequency: Float[Array, " n_dim"], params: dict[str, Float] + def __call__( + self, frequency: Float[Array, " n_dim"], params: dict[str, Float] ) -> dict[str, Float[Array, " n_dim"]]: output = {}