Updated watch agent script to remove rendering option

watch_agent.py

@@ -30,14 +30,8 @@ parser.add_argument("-e", "--num_episodes", help="specify the number of episodes
 parser.add_argument("-a", "--agent_filepath", help="file path to agent to watch, minus the .zip extension", type=str, required=True)
 args = parser.parse_args()
 
-MODEL_NAME = args.agent_filepath
-loaded_model = DQN.load(MODEL_NAME)
-
-# Toggle the render mode based on the -o flag
-# if args.observe == True:
-#     mode = "human"
-# else:
-#     mode = "rgb_array"
+model_name = args.agent_filepath
+model = DQN.load(model_name)
 
 # Retrieve the environment
 eval_env = Monitor(gym.make("ALE/Pacman-v5", 
@@ -49,10 +43,5 @@ if args.print == True:
     env_info = str(eval_env.spec).split(", ")
     for item in env_info:
         print(item)
-# Evaluate the policy
-# mean_rwd, std_rwd = 
-
-evaluate_policy(loaded_model.policy, eval_env, n_eval_episodes=args.num_episodes)
-# print("eval episodes: ", args.num_episodes)
-# print("mean rwd: ", mean_rwd)
-# print("std rwd: ", std_rwd)
+# Evaluate the policy to render the agent
+evaluate_policy(model.policy, eval_env, n_eval_episodes=args.num_episodes)