diff --git a/nevergrad/optimization/metamodel.py b/nevergrad/optimization/metamodel.py index 7cbdd2e5b..fdb33733a 100644 --- a/nevergrad/optimization/metamodel.py +++ b/nevergrad/optimization/metamodel.py @@ -174,7 +174,8 @@ def my_kernel(x, y): DiscreteLenglerOnePlusOne = registry["DiscreteLenglerOnePlusOne"] def loss_function_sm(x): - return float(model.predict(trans(np.asarray(x, dtype=X[0].dtype).flatten()[None, :]))) + x = np.asarray(x, dtype=X[0].dtype).flatten()[None, :] + return model.predict(trans(x)).item() for cls in ( (Powell, DE) if algorithm != "image" else (DiscreteLenglerOnePlusOne,)