code: def data_generator(data):
      for elem in data:
          yield np.array(elem).reshape((1, 3))
  
file_path: AmpliGraph/tests/ampligraph/datasets/test_data_indexer.py
imports: from ampligraph.datasets.data_indexer import DataIndexer
  import numpy
  import pytest
label_desc: Use image node_id 1 for calling a global function with example usage: data_generator(data) without return types