From 6a5687293200233934eda8f75395f0d218399b71 Mon Sep 17 00:00:00 2001 From: Martin Bubel Date: Tue, 17 Oct 2023 08:30:52 +0200 Subject: [PATCH] fix pytesting minibatch --- GPy/testing/test_minibatch.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/GPy/testing/test_minibatch.py b/GPy/testing/test_minibatch.py index 96ab617f..34e325d6 100644 --- a/GPy/testing/test_minibatch.py +++ b/GPy/testing/test_minibatch.py @@ -52,7 +52,8 @@ class TestBGPLVM: ) np.testing.assert_allclose(m.gradient, self.m_full.gradient) - self.assertRaises(NotImplementedError, m.predict, m.X, full_cov=True) + with pytest.raises(NotImplementedError): + m.predict(m.X, full_cov=True) mu1, var1 = m.predict(m.X, full_cov=False) mu2, var2 = self.m_full.predict(self.m_full.X, full_cov=False) @@ -124,7 +125,7 @@ class TestBGPLVM: assert m.checkgrad() def test_gradients_missingdata(self): - self.seutp() + self.setup() m = GPy.models.bayesian_gplvm_minibatch.BayesianGPLVMMiniBatch( self.Y, self.Q,