fix: typing

zfit · Nov 1, 2024 · 7bdb6cc · 7bdb6cc
1 parent 99684eb
commit 7bdb6cc
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/zfit_physics/models/pdf_hypatia2.py b/src/zfit_physics/models/pdf_hypatia2.py
@@ -1,5 +1,7 @@
 from __future__ import annotations
 
+from collections.abc import Mapping
+
 import numpy as np
 import tensorflow as tf
 import tensorflow_probability as tfp
@@ -322,7 +324,7 @@ def __init__(
         )
 
     @zfit.supports(norm=False)
-    def _pdf(self, x: tf.Tensor, norm, params) -> tf.Tensor:
+    def _pdf(self, x: Mapping[int, tf.Tensor], norm, params: Mapping[str, tf.Tensor]) -> tf.Tensor:
         del norm
         x0 = x[0]
         mu = params["mu"]