code: def setUp(self) -> None:
      self.exp = get_branin_experiment(with_batch=True)
      self.exp.trials[0].run()
      self.model = Models.BOTORCH_MODULAR(
          # Model bridge kwargs
          experiment=self.exp,
          data=self.exp.fetch_data(),
      )
  
file_path: Ax/ax/plot/tests/test_traces.py
imports: import numpy
  import plotly.graph_objects
  from ax.modelbridge.registry import Models
  from ax.plot.base import AxPlotConfig
  from ax.plot.trace import optimization_trace_single_method, optimization_trace_single_method_plotly, plot_objective_value_vs_trial_index
  from ax.service.utils.report_utils import exp_to_df
  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 TracesTest obj's underlying member method code with example usage: obj.setUp() without return types