code: def get_mock_wandb_logger(mock_api_cls=_MockWandbAPI, **kwargs):
      class MockWandbLoggingActor(_MockWandbLoggingActor):
          _mock_wandb_api_cls = mock_api_cls
  
      logger = WandbTestExperimentLogger(
          project="test_project",
          api_key="1234",
          **kwargs,
      )
      logger._logger_actor_cls = MockWandbLoggingActor
      return logger
  
imports: from collections import namedtuple
  from dataclasses import dataclass
  from typing import Dict
  from unittest.mock import Mock
  from wandb.util import json_dumps_safer
  import ray
  from ray.air.integrations.wandb import _WandbLoggingActor, WandbLoggerCallback
label_desc: Use image node_id 1 for calling a global function with example usage: get_mock_wandb_logger(mock_api_cls) and returns: logger