diff --git a/GPy/testing/pickle_tests.py b/GPy/testing/pickle_tests.py index bf00055f..59818de7 100644 --- a/GPy/testing/pickle_tests.py +++ b/GPy/testing/pickle_tests.py @@ -23,6 +23,7 @@ from functools import reduce from GPy.util.caching import Cacher import GPy from pickle import PicklingError +import GPy def toy_model(): X = np.linspace(0,1,50)[:, None]