class_docstr: Class to interface with the Ax Platform
class_docstr_tok: ["Class","to","interface","with","the","Ax","Platform"]
code: def __init__(
      self, ax_config: AxConfig, max_batch_size: Optional[int]
  ):
      self.config: Optional[DictConfig] = None
      self.launcher: Optional[Launcher] = None
      self.hydra_context: Optional[HydraContext] = None
  
      self.job_results = None
      self.experiment: ExperimentConfig = ax_config.experiment
      self.early_stopper: EarlyStopper = EarlyStopper(
          max_epochs_without_improvement=ax_config.early_stop.max_epochs_without_improvement,
          epsilon=ax_config.early_stop.epsilon,
          minimize=ax_config.early_stop.minimize,
      )
      self.ax_client_config: ClientConfig = ax_config.client
      self.max_trials = ax_config.max_trials
      self.ax_params: DictConfig = OmegaConf.create({})
      if hasattr(ax_config, "params"):
          self.ax_params.update(ax_config.params)
      self.sweep_dir: str
      self.job_idx: Optional[int] = None
      self.max_batch_size = max_batch_size
      self.is_noisy: bool = ax_config.is_noisy
  
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 1 to create a new CoreAxSweeper object from inherited base classes: Sweeper with example: obj = CoreAxSweeper(ax_config, max_batch_size)