code: def _parse_dtype(self, space):
      
      if isinstance(space, gym.spaces.Discrete):
        return tf.int32
      if isinstance(space, gym.spaces.Box):
        return tf.float32
      raise NotImplementedError()
docs: Get a tensor dtype from a OpenAI Gym space.
  
      Args:
        space: Gym space.
  
      Raises:
        NotImplementedError: For spaces other than Box and Discrete.
  
      Returns:
        TensorFlow data type.