@@ -53,7 +53,6 @@ def disaster_model():
53
53
return disaster_model , years
54
54
55
55
56
- @pytest .mark .filterwarnings ("error" )
57
56
def test_marginalized_bernoulli_logp ():
58
57
"""Test logp of IR TestFiniteMarginalDiscreteRV directly"""
59
58
mu = pt .vector ("mu" )
@@ -83,7 +82,6 @@ def test_marginalized_bernoulli_logp():
83
82
)
84
83
85
84
86
- @pytest .mark .filterwarnings ("error" )
87
85
def test_marginalized_basic ():
88
86
data = [2 ] * 5
89
87
@@ -128,7 +126,6 @@ def test_marginalized_basic():
128
126
)
129
127
130
128
131
- @pytest .mark .filterwarnings ("error" )
132
129
def test_multiple_independent_marginalized_rvs ():
133
130
with MarginalModel () as m :
134
131
sigma = pm .HalfNormal ("sigma" )
@@ -155,7 +152,6 @@ def test_multiple_independent_marginalized_rvs():
155
152
np .testing .assert_array_almost_equal (y_logp , y_ref_logp )
156
153
157
154
158
- @pytest .mark .filterwarnings ("error" )
159
155
def test_multiple_dependent_marginalized_rvs ():
160
156
"""Test that marginalization works when there is more than one dependent RV"""
161
157
with MarginalModel () as m :
@@ -194,7 +190,6 @@ def test_rv_dependent_multiple_marginalized_rvs():
194
190
np .testing .assert_allclose (np .exp (logp ({"z" : 2 })), 0.1 * 0.3 )
195
191
196
192
197
- @pytest .mark .filterwarnings ("error" )
198
193
def test_nested_marginalized_rvs ():
199
194
"""Test that marginalization works when there are nested marginalized RVs"""
200
195
@@ -232,7 +227,6 @@ def test_nested_marginalized_rvs():
232
227
)
233
228
234
229
235
- @pytest .mark .filterwarnings ("error" )
236
230
def test_marginalized_change_point_model (disaster_model ):
237
231
m , years = disaster_model
238
232
@@ -251,7 +245,6 @@ def test_marginalized_change_point_model(disaster_model):
251
245
252
246
253
247
@pytest .mark .slow
254
- @pytest .mark .filterwarnings ("error" )
255
248
def test_marginalized_change_point_model_sampling (disaster_model ):
256
249
m , _ = disaster_model
257
250
@@ -435,7 +428,6 @@ def true_sub_idx_logp(y):
435
428
np .testing .assert_almost_equal (logsumexp (post .lp_sub_idx , axis = - 1 ), 0 )
436
429
437
430
438
- @pytest .mark .filterwarnings ("error" )
439
431
def test_not_supported_marginalized ():
440
432
"""Marginalized graphs with non-Elemwise Operations are not supported as they
441
433
would violate the batching logp assumption"""
@@ -484,7 +476,6 @@ def test_not_supported_marginalized():
484
476
m .marginalize (x )
485
477
486
478
487
- @pytest .mark .filterwarnings ("error" )
488
479
def test_marginalized_deterministic_and_potential ():
489
480
rng = np .random .default_rng (299 )
490
481
@@ -510,7 +501,6 @@ def test_marginalized_deterministic_and_potential():
510
501
assert pot_value .eval ({y_value : 2 , z_value : 5 }) == 8
511
502
512
503
513
- @pytest .mark .filterwarnings ("error" )
514
504
def test_not_supported_marginalized_deterministic_and_potential ():
515
505
with MarginalModel () as m :
516
506
x = pm .Bernoulli ("x" , p = 0.7 )
@@ -533,7 +523,6 @@ def test_not_supported_marginalized_deterministic_and_potential():
533
523
m .marginalize ([x ])
534
524
535
525
536
- @pytest .mark .filterwarnings ("error" )
537
526
@pytest .mark .parametrize (
538
527
"transform, expected_warning" ,
539
528
(
0 commit comments