code: def check_close(xs, ys, atol=None, rtol=None, err_msg=""):
      assert_close = partial(
          _assert_numpy_close, atol=atol, rtol=rtol, err_msg=err_msg
      )
      tree_map(assert_close, xs, ys)
  
imports: from functools import partial
  import operator
  from jax._src import api
  from jax._src import config
  from jax._src import dtypes
  from jax._src.tree_util import tree_map, tree_reduce
  import numpy
label_desc: Use image node_id 7 for calling a global function with example usage: check_close(xs, ys, atol, rtol, err_msg) without return types