Skip to content

Commit 5c2c49d

Browse files
ArmavicaricardoV94
authored andcommitted
Remove two unused fixtures
* fit_method_with_object * aevb_model
1 parent dec7d6f commit 5c2c49d

File tree

1 file changed

+0
-27
lines changed

1 file changed

+0
-27
lines changed

pymc/tests/test_variational_inference.py

Lines changed: 0 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -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-
676649
def test_pickle_approx(three_var_approx):
677650
import cloudpickle
678651

0 commit comments

Comments
 (0)