From 0d9eb72fbe145820ef27b1d4236469802dd9a6bd Mon Sep 17 00:00:00 2001 From: loopy langur Date: Sat, 23 Jan 2021 20:56:26 -0600 Subject: [PATCH] Optimizing test --- pmlearn/gaussian_process/tests/test_gpr.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pmlearn/gaussian_process/tests/test_gpr.py b/pmlearn/gaussian_process/tests/test_gpr.py index afc947b..2230ec1 100644 --- a/pmlearn/gaussian_process/tests/test_gpr.py +++ b/pmlearn/gaussian_process/tests/test_gpr.py @@ -88,7 +88,7 @@ class TestGaussianProcessRegressorPredict(TestGaussianProcessRegressor): def test_predict_returns_predictions(self): print('') self.advi_gpr.fit(self.X_train, self.y_train, - inference_args={"n": 25000}) + inference_args={"n": 200}) preds = self.advi_gpr.predict(self.X_test) npt.assert_equal(self.y_test.shape, preds.shape)