diff --git a/smt/applications/tests/test_ego.py b/smt/applications/tests/test_ego.py index 8567edfd7..2622989a1 100644 --- a/smt/applications/tests/test_ego.py +++ b/smt/applications/tests/test_ego.py @@ -854,7 +854,7 @@ def _evaluate(self, x, kx): assert kx is None response = self.super._evaluate(x, kx) sens = np.hstack( - self.super._evaluate(x, ki) for ki in range(x.shape[1]) + [self.super._evaluate(x, ki) for ki in range(x.shape[1])] ) return np.hstack((response, sens))