code: def single_batch_input_fn(features, labels=None):
      def input_fn():
          inputs = features if labels is None else (features, labels)
          dataset = tf.data.Dataset.from_tensor_slices(inputs)
          return dataset.batch(len(features))
  
      return input_fn
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import os
  import shutil
  import tempfile
  from absl.testing import parameterized
  import neural_structured_learning.configs
  import neural_structured_learning.estimator
  import numpy
  import tensorflow
  from tensorflow import estimator
  from tensorflow.python.framework import test_util
label_desc: Use image node_id 1 for calling a global function with example usage: single_batch_input_fn(features, labels) and returns: input_fn, dataset