Skip to content

Commit bf5fd24

Browse files
authored
fix typo (#2552)
1 parent 67e9db9 commit bf5fd24

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

pymc3/tests/test_variational_inference.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -685,20 +685,20 @@ def test_aevb(inference_spec, aevb_model):
685685
pytest.skip('Does not support AEVB')
686686

687687

688-
def test_batched_approx(three_var_model, parametric_grouped_approxes):
688+
def test_rowwise_approx(three_var_model, parametric_grouped_approxes):
689689
# add to inference that supports aevb
690690
cls, kw = parametric_grouped_approxes
691691
with three_var_model:
692692
try:
693-
approx = Approximation([cls([three_var_model.one], batched=True, **kw), Group(None, vfam='mf')])
693+
approx = Approximation([cls([three_var_model.one], rowwise=True, **kw), Group(None, vfam='mf')])
694694
inference = pm.KLqp(approx)
695695
approx = inference.fit(3, obj_n_mc=2)
696696
approx.sample(10)
697697
approx.sample_node(
698698
three_var_model.one
699699
).eval()
700700
except pm.opvi.BatchedGroupError:
701-
pytest.skip('Does not support grouping')
701+
pytest.skip('Does not support rowwise grouping')
702702

703703

704704
@pytest.fixture('module')

0 commit comments

Comments
 (0)