From 221f7f792f4da943261fed106498c3e7d2adda51 Mon Sep 17 00:00:00 2001 From: Alex Feldstein Date: Fri, 17 Mar 2017 19:32:47 -0400 Subject: [PATCH] fix for parallel optimization --- GPy/core/gp.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/GPy/core/gp.py b/GPy/core/gp.py index b90c95c1..7f23e5af 100644 --- a/GPy/core/gp.py +++ b/GPy/core/gp.py @@ -562,11 +562,12 @@ class GP(Model): """ self.inference_method.on_optimization_start() try: - super(GP, self).optimize(optimizer, start, messages, max_iters, ipython_notebook, clear_after_finish, **kwargs) + ret = super(GP, self).optimize(optimizer, start, messages, max_iters, ipython_notebook, clear_after_finish, **kwargs) except KeyboardInterrupt: print("KeyboardInterrupt caught, calling on_optimization_end() to round things up") self.inference_method.on_optimization_end() raise + return ret def infer_newX(self, Y_new, optimize=True): """