Skip to content

Commit e74cf7f

Browse files
committed
FIX: set correct builder signature
1 parent ad6c364 commit e74cf7f

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

src/ampform_dpd/dynamics/builder.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323

2424
def formulate_breit_wigner_with_form_factor(
2525
decay: ThreeBodyDecayChain,
26-
) -> tuple[sp.Expr, dict[sp.Symbol, float]]:
26+
) -> tuple[sp.Expr, dict[sp.Symbol, complex | float]]:
2727
decay_node = decay.decay_node
2828
s = get_mandelstam_s(decay_node)
2929
parameter_defaults = {}
@@ -41,7 +41,7 @@ def formulate_breit_wigner_with_form_factor(
4141

4242
def _create_form_factor(
4343
s: sp.Symbol, isobar: IsobarNode
44-
) -> tuple[sp.Expr, dict[sp.Symbol, float]]:
44+
) -> tuple[sp.Expr, dict[sp.Symbol, complex | float]]:
4545
if isinstance(isobar.parent, State):
4646
inv_mass = sp.Symbol("m0", nonnegative=True)
4747
else:
@@ -56,7 +56,7 @@ def _create_form_factor(
5656
angular_momentum=_get_angular_momentum(isobar),
5757
meson_radius=meson_radius,
5858
)
59-
parameter_defaults = {
59+
parameter_defaults: dict[sp.Symbol, complex | float] = {
6060
meson_radius: 1,
6161
outgoing_state_mass1: to_particle(isobar.child1).mass,
6262
outgoing_state_mass2: to_particle(isobar.child2).mass,
@@ -68,7 +68,7 @@ def _create_form_factor(
6868

6969
def _create_breit_wigner(
7070
s: sp.Symbol, isobar: DecayNode
71-
) -> tuple[sp.Expr, dict[sp.Symbol, float]]:
71+
) -> tuple[sp.Expr, dict[sp.Symbol, complex | float]]:
7272
outgoing_state_mass1 = create_mass_symbol(isobar.child1)
7373
outgoing_state_mass2 = create_mass_symbol(isobar.child2)
7474
angular_momentum = _get_angular_momentum(isobar)
@@ -85,7 +85,7 @@ def _create_breit_wigner(
8585
angular_momentum=angular_momentum,
8686
meson_radius=meson_radius,
8787
)
88-
parameter_defaults = {
88+
parameter_defaults: dict[sp.Symbol, complex | float] = {
8989
res_mass: isobar.parent.mass,
9090
res_width: isobar.parent.width,
9191
meson_radius: 1,

0 commit comments

Comments
 (0)