code: def mock_2x2_devices():
      """Hard-coded reproduction of jax.devices() output on v3-2x2."""
      return mock_tpu_devices(2, 2, 1, "TPU v3", False)
  
imports: import collections
  from collections.abc import Sequence
  import dataclasses
  import numpy
  from absl import logging
  from absl.testing import absltest
  from absl.testing import parameterized
  from jax.experimental import mesh_utils
  from jax.sharding import Mesh
  from jax._src import test_util
label_desc: Use image node_id 3 for calling a global function with example usage: mock_2x2_devices() and returns: mock_tpu_devices