diff --git a/ax/models/random/sobol.py b/ax/models/random/sobol.py index 9e62ce868ec..a2eb6aa6b16 100644 --- a/ax/models/random/sobol.py +++ b/ax/models/random/sobol.py @@ -119,7 +119,7 @@ def gen( ) if self.engine: self.init_position = not_none(self.engine).num_generated - return (points, weights) + return points, weights # pyre-fixme[24]: Generic type `np.ndarray` expects 2 type parameters. def _gen_samples(self, n: int, tunable_d: int) -> np.ndarray: