class_docstr: Wrapper class for MinAtar environments with sticky actions.
class_docstr_tok: ["Wrapper","class","for","MinAtar","environments","with","sticky","actions","."]
code: def step(self, action):
      action_to_use = action
      if self._prev_action:
          self._rng, rng = jax.random.split(self._rng, num=2)
          if jax.random.uniform(rng) < self.action_repeat_probability:
              action_to_use = self._prev_action
      self._prev_action = action
      return super().step(action_to_use)
  
imports: import time
  from dopamine.discrete_domains import atari_lib
  from flax import linen
  import gin
  import jax
  import jax.numpy
  import minatar
label_desc: Use image node_id 3 for calling the StickyMinAtarEnv obj's underlying member method code with example usage: obj.step(action) and returns: super