Update waveform.py

Update typing information

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: Array, params: dict) -> dict:
+    def __call__(self, frequency: Float[Array, " n_dim"], params: dict[str, Float]
+    ) -> dict[str, Float[Array, " n_dim"]]
         output = {}
 
         if self.use_lambda_tildes:
@@ -151,7 +152,8 @@ def __init__(
         self.use_lambda_tildes = use_lambda_tildes
         self.no_taper = no_taper
 
-    def __call__(self, frequency: Array, params: dict) -> dict:
+    def __call__(self, frequency: Float[Array, " n_dim"], params: dict[str, Float]
+    ) -> dict[str, Float[Array, " n_dim"]]
         output = {}
 
         if self.use_lambda_tildes: