@@ -614,20 +614,6 @@ def another_simple_model():
614614 return _model
615615
616616
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-
631617@pytest .mark .parametrize (
632618 ["method" , "kwargs" , "error" ],
633619 [
@@ -660,19 +646,6 @@ def test_fit_fn_text(method, kwargs, error, another_simple_model):
660646 fit (10 , method = method , ** kwargs )
661647
662648
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-
676649def test_pickle_approx (three_var_approx ):
677650 import cloudpickle
678651
0 commit comments