@@ -540,38 +540,12 @@ def test_normal_scalar_idata(self):
540
540
ppc = pm .sample_posterior_predictive (idata , return_inferencedata = False )
541
541
assert ppc ["a" ].shape == (nchains , ndraws )
542
542
543
- def test_external_trace (self ):
544
- nchains = 2
545
- ndraws = 500
546
- with pm .Model () as model :
547
- mu = pm .Normal ("mu" , 0.0 , 1.0 )
548
- a = pm .Normal ("a" , mu = mu , sigma = 1 , observed = 0.0 )
549
- trace = pm .sample (
550
- draws = ndraws ,
551
- chains = nchains ,
552
- )
553
-
554
- # test that trace is used in ppc
555
- with pm .Model () as model_ppc :
556
- mu = pm .Normal ("mu" , 0.0 , 1.0 )
557
- a = pm .Normal ("a" , mu = mu , sigma = 1 )
558
-
559
- ppc = pm .sample_posterior_predictive (
560
- trace = trace , model = model_ppc , return_inferencedata = False
561
- )
562
- assert list (ppc .keys ()) == ["a" ]
563
-
564
543
def test_external_trace_det (self ):
565
- nchains = 2
566
- ndraws = 500
567
544
with pm .Model () as model :
568
545
mu = pm .Normal ("mu" , 0.0 , 1.0 )
569
546
a = pm .Normal ("a" , mu = mu , sigma = 1 , observed = 0.0 )
570
547
b = pm .Deterministic ("b" , a + 1 )
571
- trace = pm .sample (
572
- draws = ndraws ,
573
- chains = nchains ,
574
- )
548
+ trace = pm .sample (tune = 50 , draws = 50 , chains = 1 , compute_convergence_checks = False )
575
549
576
550
# test that trace is used in ppc
577
551
with pm .Model () as model_ppc :
0 commit comments