Fix stochastic optimization (#328)

* Fix incompatibility with scipy>1.5 * Modified CHANGELOG
elfi-dev · Jul 29, 2020 · ec1418e · ec1418e
1 parent 2f049aa
commit ec1418e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/CHANGELOG.rst b/CHANGELOG.rst
@@ -1,6 +1,7 @@
 Changelog
 =========
 
+- Fix incompatibility with scipy>1.5 in bo.utils.stochastic_optimization
 - Minor improvements to documentation
 
 0.7.5 (2019-12-18)

diff --git a/elfi/methods/bo/utils.py b/elfi/methods/bo/utils.py
@@ -28,8 +28,12 @@ def stochastic_optimization(fun, bounds, maxiter=1000, polish=True, seed=0):
     tuple of the found coordinates of minimum and the corresponding value.
 
     """
+    def fun_1d(x):
+        return fun(x).ravel()
+
     result = differential_evolution(
-        func=fun, bounds=bounds, maxiter=maxiter, polish=polish, init='latinhypercube', seed=seed)
+        func=fun_1d, bounds=bounds, maxiter=maxiter,
+        polish=polish, init='latinhypercube', seed=seed)
     return result.x, result.fun