@@ -540,38 +540,12 @@ def test_normal_scalar_idata(self):
540540 ppc = pm .sample_posterior_predictive (idata , return_inferencedata = False )
541541 assert ppc ["a" ].shape == (nchains , ndraws )
542542
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-
564543 def test_external_trace_det (self ):
565- nchains = 2
566- ndraws = 500
567544 with pm .Model () as model :
568545 mu = pm .Normal ("mu" , 0.0 , 1.0 )
569546 a = pm .Normal ("a" , mu = mu , sigma = 1 , observed = 0.0 )
570547 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 )
575549
576550 # test that trace is used in ppc
577551 with pm .Model () as model_ppc :
0 commit comments