Added a function to check the agent's average return for a given map.