code: def __init__(
      self,
      max_trials: int = 30,
      with_fidelity: bool = False,
      shape: tuple[int, ...] = (2, 1),
  ):
      super().__init__(max_trials, with_fidelity)
      self.shape = shape
  
imports: from __future__ import annotations
  import copy
  import os
  from pathlib import Path
  import numpy
  import pytest
  from orion.algo.pbt.pb2_utils import import_optional
  from orion.client import build_experiment, create_experiment, workon
  from orion.core.io.space_builder import SpaceBuilder
  from orion.core.utils.module_import import ImportOptional
  from orion.core.worker.algo_wrappers import AlgoWrapper
  from orion.core.worker.primary_algo import SpaceTransform
  from orion.core.worker.trial import Trial
  from orion.storage.base import BaseStorageProtocol
  from orion.benchmark.task.base import BenchmarkTask
label_desc: Use image node_id 1 to create a new MultiDimRosenbrock object from inherited base classes: CustomRosenbrock with example: obj = MultiDimRosenbrock(max_trials, with_fidelity, shape)