code: def setUp(self) -> None:
      self.experiment = get_branin_experiment()
      sobol = Models.SOBOL(search_space=self.experiment.search_space)
      sobol_run = sobol.gen(n=20)
      self.experiment.new_batch_trial().add_generator_run(
          sobol_run
      ).run().mark_completed()
      self.data = self.experiment.fetch_data()
  
imports: import warnings
  from unittest import mock
  from unittest.mock import MagicMock, Mock, patch
  from ax.core.observation import ObservationFeatures
  from ax.exceptions.core import UserInputError
  from ax.modelbridge.factory import get_sobol
  from ax.modelbridge.model_spec import FactoryFunctionModelSpec, ModelSpec
  from ax.modelbridge.modelbridge_utils import extract_search_space_digest
  from ax.modelbridge.registry import Models
  from ax.utils.common.testutils import TestCase
  from ax.utils.testing.core_stubs import get_branin_experiment
  from ax.utils.testing.mock import fast_botorch_optimize
label_desc: Use image node_id 1 for calling the BaseModelSpecTest obj's underlying member method code with example usage: obj.setUp() without return types