code: def _assert_arrays_equal(x: jax.Array, y: jax.Array) -> None:
      np.testing.assert_almost_equal(
          np.array(x), np.array(y), decimal=5
      )
  
imports: import inspect
  from typing import Optional, Union
  from absl.testing import absltest
  from absl.testing import parameterized
  from haiku._src import base
  from haiku._src import multi_transform
  from haiku._src import transform
  from haiku._src import typing
  import jax
  import jax.numpy
  import numpy
label_desc: Use image node_id 1 for calling a global function with example usage: _assert_arrays_equal(x, y) without return types