skip specific cases on the top level

Author: Maxim Kochurov

pymc/tests/variational/test_inference.py

@@ -149,14 +149,13 @@ def fit_kwargs(inference, use_minibatch):
         inference.approx.scale_cost_to_minibatch = False
     else:
         key = "full"
+    if (type(inference), key) in {(SVGD, "mini"), (ASVGD, "mini")}:
+        pytest.skip("Not Implemented Inderence")
     return _select[(type(inference), key)]
 
 
 def test_fit_oo(inference, fit_kwargs, simple_model_data):
-    try:
-        trace = inference.fit(**fit_kwargs).sample(10000)
-    except opvi.NotImplementedInference as e:
-        pytest.skip(str(e))
+    trace = inference.fit(**fit_kwargs).sample(10000)
     mu_post = simple_model_data["mu_post"]
     d = simple_model_data["d"]
     np.testing.assert_allclose(np.mean(trace.posterior["mu"]), mu_post, rtol=0.05)