From 2a01e9008a6414feca031032f7b15109236e7d46 Mon Sep 17 00:00:00 2001 From: Max Zwiessele Date: Sat, 10 Oct 2015 23:21:41 +0100 Subject: [PATCH] Update pickle_tests.py --- GPy/testing/pickle_tests.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/GPy/testing/pickle_tests.py b/GPy/testing/pickle_tests.py index d4d96a24..ad3c4f0b 100644 --- a/GPy/testing/pickle_tests.py +++ b/GPy/testing/pickle_tests.py @@ -22,6 +22,7 @@ from GPy.models.gp_regression import GPRegression from functools import reduce from GPy.util.caching import Cacher from pickle import PicklingError +import GPy def toy_model(): X = np.linspace(0,1,50)[:, None] @@ -64,8 +65,7 @@ class Test(ListDictTestCase): with f: pickle.dump(piov, f) - f.seek(0) - + pio2 = GPy.load(f) #py3 fix #self.assertListDictEquals(dict(piov.items()), dict(pio2.iteritems()))