class_docstr: 1-dimensional grid-world implementation with 3 possible movement actions ("LEFT", "STAY", "RIGHT")
code: def reset(self, init_state=None):
      self.set_init_state(init_state)
      self.last_action = None
      return self.state
  
imports: import numpy
  import matplotlib.pyplot
  import seaborn
  from pymdp.envs import Env
label_desc: Use image node_id 2 for calling the DGridWorldEnv obj's underlying member method code with example usage: obj.reset(init_state) and returns: self