Skip to content

Commit 25c0ee3

Browse files
committed
fix: typing
1 parent 62fcad6 commit 25c0ee3

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/zfit_physics/models/pdf_hypatia2.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
11
from __future__ import annotations
22

3+
from collections.abc import Mapping
4+
35
import numpy as np
46
import tensorflow as tf
57
import tensorflow_probability as tfp
@@ -322,7 +324,7 @@ def __init__(
322324
)
323325

324326
@zfit.supports(norm=False)
325-
def _pdf(self, x: tf.Tensor, norm, params) -> tf.Tensor:
327+
def _pdf(self, x: Mapping[int, tf.Tensor], norm, params: Mapping[str, tf.Tensor]) -> tf.Tensor:
326328
del norm
327329
x0 = x[0]
328330
mu = params["mu"]

0 commit comments

Comments
 (0)