class_docstr: 2-dimensional grid-world implementation with 5 actions (the 4 cardinal directions and staying put).
code: def get_init_state_dist(self, init_state=None):
      init_state_dist = np.zeros(self.n_states)
      if init_state == None:
          init_state_dist[self.init_state] = 1.0
      else:
          init_state_dist[init_state] = 1.0
  
imports: import numpy
  import matplotlib.pyplot
  import seaborn
  from pymdp.envs import Env
label_desc: Use image node_id 8 for calling the GridWorldEnv obj's underlying member method code with example usage: obj.get_init_state_dist(init_state) without return types