code: def storage(tmp_path):
      os.makedirs(tmp_path / "exp_name" / "trial_name", exist_ok=True)
      yield StorageContext(
          storage_path=str(tmp_path),
          experiment_dir_name="exp_name",
          trial_dir_name="trial_name",
      )
  
imports: import time
  import os
  import pytest
  import sys
  from unittest.mock import MagicMock
  import ray
  from ray import tune
  from ray.train import CheckpointConfig
  from ray.cluster_utils import Cluster
  from ray.train._internal.storage import StorageContext
  from ray.tune.error import TuneError
  from ray.tune.search import BasicVariantGenerator
  from ray.tune.experiment import Trial
  from ray.tune.execution.tune_controller import TuneController
label_desc: Use image node_id 6 for calling a global function with example usage: storage(tmp_path) without return types