class_docstr: Env wrapper that marks bad_transition
      
class_docstr_tok: ["Env","wrapper","that","marks","bad_transition"]
code: def step(self, action):
      obs, rew, done, info = self.env.step(action)
      if (
          done
          and self.env._max_episode_steps == self.env._elapsed_steps
      ):
          info["bad_transition"] = True
      return obs, rew, done, info
  
imports: import numpy
  import gym
  import time
  from parl.env.compat_wrappers import CompatWrapper
label_desc: Use image node_id 1 for calling the TimeLimitMaskEnv obj's underlying member method code with example usage: obj.step(action) and returns: obs,  rew,  done,  info