class_docstr: Raise an error after specified number of steps in an episode.
class_docstr_tok: ["Raise","an","error","after","specified","number","of","steps","in","an","episode","."]
code: def _step(self, *args, **kwargs):
      transition = super(MockEnvironmentCrashInStep, self)._step(
          *args, **kwargs
      )
      self._steps += 1
      if self._steps == self._crash_at_step:
          raise RuntimeError()
      return transition
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import collections
  import functools
  import time
  import numpy
  import tensorflow
  from tf_agents.environments import parallel_py_environment
  from tf_agents.environments import py_environment
  from tf_agents.environments import random_py_environment
  from tf_agents.specs import array_spec
  from tf_agents.system import system_multiprocessing
  from tf_agents.trajectories import time_step
label_desc: Use image node_id 2 for calling the MockEnvironmentCrashInStep obj's underlying member method code with example usage: obj._step() and returns: transition