code: def __init__(self):
      self.batch = 10
      self.timestep = 5
      self.memory_size = 6
      self.units = 8
  
      self.memory = np.random.randn(
          self.batch, self.timestep, self.memory_size
      ).astype(np.float32)
      self.memory_length = np.random.randint(
          low=1, high=self.timestep + 1, size=(self.batch,)
      )
      self.query = np.random.randn(self.batch, self.units).astype(
          np.float32
      )
      self.state = np.random.randn(self.batch, self.timestep).astype(
          np.float32
      )
  
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 1 to create a new DummyData object  with example: obj = DummyData()