code: def discrete_space_spec(discrete_space, tf_dtype):
      del discrete_space  # this is not needed.
      return tf.FixedLenFeature((1,), tf_dtype)
  
file_path: tensor2tensor/tensor2tensor/envs/gym_spaces_utils.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from gym.spaces import Box
  from gym.spaces import Discrete
  import numpy
  import tensorflow.compat.v1
label_desc: Use image node_id 2 for calling a global function with example usage: discrete_space_spec(discrete_space, tf_dtype) and returns: tf