code: def ray_start_2_cpus():
      address_info = ray.init(num_cpus=2)
      yield address_info
      # The code after the yield will run as teardown code.
      ray.shutdown()
  
imports: import json
  import os
  from typing import Dict, Union
  import pytest
  import ray
  from ray import train, tune
  from ray.train._internal.storage import StorageContext
  from ray.tune.trainable import wrap_function
  from ray.train.tests.util import create_dict_checkpoint
label_desc: Use image node_id 1 for calling a global function with example usage: ray_start_2_cpus() without return types