code: def test_tag_setup_wandb(mock_record):
      from ray.air.integrations.wandb import _setup_wandb
  
      with patch.dict(
          os.environ, {wandb.WANDB_MODE_ENV_VAR: "disabled"}
      ):
          _setup_wandb(
              trial_id="a",
              trial_name="b",
              config={},
              _wandb=MagicMock(),
          )
      assert mock_record[TagKey.AIR_SETUP_WANDB_INTEGRATION_USED] == "1"
  
imports: import json
  import os
  from packaging.version import Version
  import pyarrow.fs
  import pytest
  from unittest.mock import MagicMock, patch
  import ray
  from ray import train, tune
  from ray.air._internal import usage
  from ray.air._internal.usage import AirEntrypoint
  from ray.air.integrations import wandb, mlflow, comet
  from ray.train._internal.storage import StorageContext
  from ray.tune.callback import Callback
  from ray.tune.experiment.experiment import Experiment
  from ray.tune.logger import LoggerCallback
  from ray.tune.logger.aim import AimLoggerCallback
  from ray.tune.utils.callback import DEFAULT_CALLBACK_CLASSES
  from ray._private.usage.usage_lib import TagKey
label_desc: Use image node_id 8 for calling a global function with example usage: test_tag_setup_wandb(mock_record) without return types