code: def assert_allclose_or_equal(x, y, **kwargs):
      if isinstance(x, np.ndarray) or isinstance(x, float):
          np.testing.assert_allclose(x, y, atol=1e-3, **kwargs)
      else:
          assert x == y
  
imports: import collections
  from packaging.version import Version
  import pytest
  import numpy
  import tensorflow
  from tensorflow_addons.seq2seq import attention_wrapper
  from tensorflow_addons.seq2seq import basic_decoder
  from tensorflow_addons.seq2seq import sampler
label_desc: Use image node_id 10 for calling a global function with example usage: assert_allclose_or_equal(x, y) without return types