class_docstr: Class to interface with the Ax Platform
class_docstr_tok: ["Class","to","interface","with","the","Ax","Platform"]
code: def setup(
      self,
      *,
      hydra_context: HydraContext,
      task_function: TaskFunction,
      config: DictConfig,
  ) -> None:
      self.config = config
      self.hydra_context = hydra_context
      self.launcher = Plugins.instance().instantiate_launcher(
          config=config,
          hydra_context=hydra_context,
          task_function=task_function,
      )
      self.sweep_dir = config.hydra.sweep.dir
  
imports: import logging
  from dataclasses import dataclass
  from typing import Any, Dict, Iterable, List, Mapping, Optional, Tuple, Union
  from ax.core import types
  from ax.exceptions.core import SearchSpaceExhausted
  from ax.service.ax_client import AxClient
  from hydra.core.override_parser.overrides_parser import OverridesParser
  from hydra.core.override_parser.types import IntervalSweep, Override, Transformer
  from hydra.core.plugins import Plugins
  from hydra.plugins.launcher import Launcher
  from hydra.plugins.sweeper import Sweeper
  from hydra.types import HydraContext, TaskFunction
  from omegaconf import DictConfig, OmegaConf
  from ._earlystopper import EarlyStopper
  from .config import AxConfig, ClientConfig, ExperimentConfig
label_desc: Use image node_id 2 for calling the CoreAxSweeper obj's underlying member method code with example usage: obj.setup() without return types