class_docstr: A simple env that takes arguments on init.
class_docstr_tok: ["A","simple","env","that","takes","arguments","on","init","."]
code: def step(self, action):
      if action == self._done_action:
          return self.observation_space.high, +1.0, True, {}
      return self.reset(), -1.0, False, {}
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import unittest
  import gym
  from gym import spaces
  import numpy
  from tensor2tensor.rl import gym_utils
  import tensorflow.compat.v1
label_desc: Use image node_id 2 for calling the EnvWithOptions obj's underlying member method code with example usage: obj.step(action) and returns: self,  unknown,  False,  dict, self,  unknown,  True,  dict