GridEnvironment5.py

#fetching agent states

def get_agent_states(self):
            
            start_state  = None
            target_state = None
            start_state = list(map(lambda x:x[0] if len(x) > 0 else None, np.where(insert_grid_map==4)))
            
            target_state = list(map(lambda x:x[0] if len(x) > 0 else None, np.where(insert_grid_map==3)))
            
            if (start_state == [None, None] or target_state == [None, None]):
                sys.exit('Start or Target state not specified')
            
            if (start_state == self.obstacle_mask):
                sys.exit('Obstacle encountered GAME OVER !!!')
                
                return start_state, target_state