code: def skip_fit_gpytorch_mll_context_manager() -> (
      Generator[None, None, None]
  ):
      """A context manager that makes `fit_gpytorch_mll` a no-op.
  
      This should only be used to speed up slow tests.
      """
      with mock.patch(
          "botorch.fit.FitGPyTorchMLL",
          side_effect=lambda *args, **kwargs: args[0],
      ) as mock_fit:
          yield
      if mock_fit.call_count < 1:
          raise AssertionError(
              "No mocks were called in the context manager. Please remove unused "
              "skip_fit_gpytorch_mll_context_manager()."
          )
  
file_path: Ax/ax/utils/testing/mock.py
imports: from contextlib import contextmanager, ExitStack
  from functools import wraps
  from typing import Any, Callable, Dict, Generator, Optional
  from unittest import mock
  from ax.models.torch.fully_bayesian import run_inference
  from botorch.fit import fit_fully_bayesian_model_nuts
  from botorch.generation.gen import minimize_with_timeout
  from botorch.optim.initializers import gen_batch_initial_conditions, gen_one_shot_kg_initial_conditions
  from scipy.optimize import OptimizeResult
  from torch import Tensor
label_desc: Use image node_id 3 for calling a global function with example usage: skip_fit_gpytorch_mll_context_manager() without return types