diff --git a/ax/analysis/plotly/tests/test_sensitivity.py b/ax/analysis/plotly/tests/test_sensitivity.py index fd87c83bc46..80fbdcecb50 100644 --- a/ax/analysis/plotly/tests/test_sensitivity.py +++ b/ax/analysis/plotly/tests/test_sensitivity.py @@ -5,7 +5,6 @@ # pyre-strict -from itertools import product from unittest.mock import patch from ax.adapter.registry import Generators diff --git a/ax/generators/random/sobol.py b/ax/generators/random/sobol.py index 3948e3e0f32..8c76ca854e3 100644 --- a/ax/generators/random/sobol.py +++ b/ax/generators/random/sobol.py @@ -66,7 +66,9 @@ def init_engine(self, n_tunable_features: int) -> SobolEngine: if not self._engine: self._engine = SobolEngine( dimension=n_tunable_features, scramble=self.scramble, seed=self.seed - ).fast_forward(self.init_position) + ) + if self.init_position > 0: + self._engine.fast_forward(self.init_position) return self._engine @property