@@ -614,20 +614,6 @@ def another_simple_model():
614
614
return _model
615
615
616
616
617
- @pytest .fixture (
618
- params = [
619
- dict (name = "advi" , kw = dict (start = {})),
620
- dict (name = "fullrank_advi" , kw = dict (start = {})),
621
- dict (name = "svgd" , kw = dict (start = {})),
622
- ],
623
- ids = lambda d : d ["name" ],
624
- )
625
- def fit_method_with_object (request , another_simple_model ):
626
- _select = dict (advi = ADVI , fullrank_advi = FullRankADVI , svgd = SVGD )
627
- with another_simple_model :
628
- return _select [request .param ["name" ]](** request .param ["kw" ])
629
-
630
-
631
617
@pytest .mark .parametrize (
632
618
["method" , "kwargs" , "error" ],
633
619
[
@@ -660,19 +646,6 @@ def test_fit_fn_text(method, kwargs, error, another_simple_model):
660
646
fit (10 , method = method , ** kwargs )
661
647
662
648
663
- @pytest .fixture (scope = "module" )
664
- def aevb_model ():
665
- with pm .Model () as model :
666
- pm .HalfNormal ("x" , size = (2 ,), total_size = 5 )
667
- pm .Normal ("y" , size = (2 ,))
668
- x = model .x
669
- y = model .y
670
- xr = model .initial_point (0 )[model .rvs_to_values [x ].name ]
671
- mu = aesara .shared (xr )
672
- rho = aesara .shared (np .zeros_like (xr ))
673
- return {"model" : model , "y" : y , "x" : x , "replace" : dict (mu = mu , rho = rho )}
674
-
675
-
676
649
def test_pickle_approx (three_var_approx ):
677
650
import cloudpickle
678
651
0 commit comments