23
23
24
24
def formulate_breit_wigner_with_form_factor (
25
25
decay : ThreeBodyDecayChain ,
26
- ) -> tuple [sp .Expr , dict [sp .Symbol , float ]]:
26
+ ) -> tuple [sp .Expr , dict [sp .Symbol , complex | float ]]:
27
27
decay_node = decay .decay_node
28
28
s = get_mandelstam_s (decay_node )
29
29
parameter_defaults = {}
@@ -41,7 +41,7 @@ def formulate_breit_wigner_with_form_factor(
41
41
42
42
def _create_form_factor (
43
43
s : sp .Symbol , isobar : IsobarNode
44
- ) -> tuple [sp .Expr , dict [sp .Symbol , float ]]:
44
+ ) -> tuple [sp .Expr , dict [sp .Symbol , complex | float ]]:
45
45
if isinstance (isobar .parent , State ):
46
46
inv_mass = sp .Symbol ("m0" , nonnegative = True )
47
47
else :
@@ -56,7 +56,7 @@ def _create_form_factor(
56
56
angular_momentum = _get_angular_momentum (isobar ),
57
57
meson_radius = meson_radius ,
58
58
)
59
- parameter_defaults = {
59
+ parameter_defaults : dict [ sp . Symbol , complex | float ] = {
60
60
meson_radius : 1 ,
61
61
outgoing_state_mass1 : to_particle (isobar .child1 ).mass ,
62
62
outgoing_state_mass2 : to_particle (isobar .child2 ).mass ,
@@ -68,7 +68,7 @@ def _create_form_factor(
68
68
69
69
def _create_breit_wigner (
70
70
s : sp .Symbol , isobar : DecayNode
71
- ) -> tuple [sp .Expr , dict [sp .Symbol , float ]]:
71
+ ) -> tuple [sp .Expr , dict [sp .Symbol , complex | float ]]:
72
72
outgoing_state_mass1 = create_mass_symbol (isobar .child1 )
73
73
outgoing_state_mass2 = create_mass_symbol (isobar .child2 )
74
74
angular_momentum = _get_angular_momentum (isobar )
@@ -85,7 +85,7 @@ def _create_breit_wigner(
85
85
angular_momentum = angular_momentum ,
86
86
meson_radius = meson_radius ,
87
87
)
88
- parameter_defaults = {
88
+ parameter_defaults : dict [ sp . Symbol , complex | float ] = {
89
89
res_mass : isobar .parent .mass ,
90
90
res_width : isobar .parent .width ,
91
91
meson_radius : 1 ,
0 commit comments