43
43
)
44
44
from pymc .distributions .shape_utils import rv_size_is_none
45
45
from pymc .initial_point import make_initial_point_fn
46
- from pymc .math import invlogit
47
46
from pymc .model import Model
48
47
49
48
@@ -667,8 +666,8 @@ def test_dirichlet_moment(a, size, expected):
667
666
np .arange (5 ),
668
667
np .arange (1 , 6 ),
669
668
(2 , 5 ),
670
- np .full ((2 , 5 ), np .arange (5 ) + np .arange (1 , 6 ) * np .euler_gamma )
671
- )
669
+ np .full ((2 , 5 ), np .arange (5 ) + np .arange (1 , 6 ) * np .euler_gamma ),
670
+ ),
672
671
],
673
672
)
674
673
def test_gumbel_moment (mu , beta , size , expected ):
@@ -689,8 +688,8 @@ def test_gumbel_moment(mu, beta, size, expected):
689
688
np .arange (- 9 , - 2 , 3 ),
690
689
np .arange (3 , 10 , 3 ),
691
690
(2 , 3 ),
692
- np .full ((2 , 3 ), np .array ([- 3 , 0 , 3 ]))
693
- )
691
+ np .full ((2 , 3 ), np .array ([- 3 , 0 , 3 ])),
692
+ ),
694
693
],
695
694
)
696
695
def test_triangular_moment (c , lower , upper , size , expected ):
@@ -706,12 +705,7 @@ def test_triangular_moment(c, lower, upper, size, expected):
706
705
(0 , np .arange (1 , 5 ), None , special .expit (np .zeros (4 ))),
707
706
(np .arange (4 ), 1 , None , special .expit (np .arange (4 ))),
708
707
(1 , 5 , 4 , special .expit (np .ones (4 ))),
709
- (
710
- np .arange (4 ),
711
- np .arange (1 , 5 ),
712
- (2 , 4 ),
713
- np .full ((2 , 4 ), special .expit (np .arange (4 )))
714
- )
708
+ (np .arange (4 ), np .arange (1 , 5 ), (2 , 4 ), np .full ((2 , 4 ), special .expit (np .arange (4 )))),
715
709
],
716
710
)
717
711
def test_logitnormal_moment (mu , sigma , size , expected ):
0 commit comments