code: def _mock_record_from_module(module, monkeypatch):
      recorded = {}
  
      def mock_record_extra_usage_tag(key: TagKey, value: str):
          recorded[key] = value
  
      monkeypatch.setattr(
          module,
          "record_extra_usage_tag",
          mock_record_extra_usage_tag,
      )
      return recorded
  
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 1 for calling a global function with example usage: _mock_record_from_module(module, monkeypatch) and returns: recorded