code: def setUp(self):
      super().setUp()
      os.environ[
          "XLA_FLAGS"
      ] = "--xla_force_host_platform_device_count=4"
      xla_bridge.get_backend.cache_clear()
      self._global_graph = _make_dynamic_batch_graph(add_globals=True)
      self._global_small_graph = _make_dynamic_batch_graph(
          add_globals=True, num_nodes=(5, 7), num_edges=(6, 8)
      )
  
imports: import functools
  import os
  from absl.testing import absltest
  from absl.testing import parameterized
  import jax
  from jax.lib import xla_bridge
  import jax.numpy
  import jax.tree_util
  from jraph._src import graph
  from jraph._src import utils
  import numpy
label_desc: Use image node_id 1 for calling the DynamicBatchTest obj's underlying member method code with example usage: obj.setUp() without return types