diff --git a/GPy/plotting/matplot_dep/plot_definitions.py b/GPy/plotting/matplot_dep/plot_definitions.py index 4d838722..1f32f346 100644 --- a/GPy/plotting/matplot_dep/plot_definitions.py +++ b/GPy/plotting/matplot_dep/plot_definitions.py @@ -55,13 +55,15 @@ class MatplotlibPlots(AbstractPlottingLibrary): if 'ax' in kwargs: ax = kwargs.pop('ax') elif 'num' in kwargs and 'figsize' in kwargs: - fig = self.figure(num=kwargs.pop('num'), figsize=kwargs.pop('figsize')).add_subplot(111, projection=projection) + fig = self.figure(num=kwargs.pop('num'), figsize=kwargs.pop('figsize')) elif 'num' in kwargs: - ax = self.figure(num=kwargs.pop('num')).add_subplot(111, projection=projection) + fig = self.figure(num=kwargs.pop('num')) elif 'figsize' in kwargs: - ax = self.figure(figsize=kwargs.pop('figsize')).add_subplot(111, projection=projection) + fig = self.figure(figsize=kwargs.pop('figsize')) else: - ax = self.figure().add_subplot(111, projection=projection) + fig = self.figure() + + ax = fig.add_subplot(self.rows, self.cols, ) return ax, kwargs @@ -83,11 +85,6 @@ class MatplotlibPlots(AbstractPlottingLibrary): ax.set_zlim(zlim) ax.figure.canvas.draw() ax.figure.show() - #try: - # ax.figure.tight_layout() - #except: - # # couldnt do tight layout, python 2.7 on MacOSX - # pass ax.figure.canvas.draw() return plots diff --git a/GPy/testing/inference_tests.py b/GPy/testing/inference_tests.py index c1fce8b9..786e5794 100644 --- a/GPy/testing/inference_tests.py +++ b/GPy/testing/inference_tests.py @@ -66,8 +66,8 @@ class InferenceXTestCase(unittest.TestCase): x, mi = m.infer_newX(m.Y) print(m.X.mean - mi.X.mean) - self.assertTrue(np.allclose(m.X.mean, mi.X.mean, rtol=1e-4, atol=1e-4)) - self.assertTrue(np.allclose(m.X.variance, mi.X.variance, rtol=1e-4, atol=1e-4)) + np.testing.assert_array_almost_equal(m.X.mean, mi.X.mean, decimal=2) + np.testing.assert_array_almost_equal(m.X.variance, mi.X.variance, decimal=2) def test_inferenceX_GPLVM(self): Ys = self.genData()