code: def create_2x2_array(scene_name, config):
      """
      Helper function for generating array of visual outcomes from the type and configuration
      """
  
      flattened_scene_array = np.array(
          ["null", "null", "null", "null"], dtype="<U6"
      )
      dot_dir1, dot_dir2 = scene_name.split("_")
      idx1, idx2 = tuple(map(lambda x: int(x) - 1, config))
  
      flattened_scene_array[idx1] = dot_dir1
      flattened_scene_array[idx2] = dot_dir2
  
      return flattened_scene_array.reshape(2, 2), flattened_scene_array
  
imports: from pymdp.envs import Env
  from pymdp import utils, maths
  import numpy
  from itertools import permutations, product
label_desc: Use image node_id 1 for calling a global function with example usage: create_2x2_array(scene_name, config) and returns: flattened_scene_array,  flattened_scene_array