Spaces:

acozma
/

CS581-Algos-Demo

Sleeping

lharri73 commited on Apr 22, 2023

Commit

3ed1eac

1 Parent(s): 1b140f8

fixed demo.py env_name

Files changed (1) hide show

demo.py CHANGED Viewed

@@ -124,7 +124,7 @@ def run(policy_fname, n_test_episodes, max_steps, render_fps, epsilon):
     agent_type, env_name = props[0], props[1]
-    agent = AGENTS_MAP[agent_type](env_name=env_name, render_mode="rgb_array")
     agent.load_policy(policy_path)
     env_action_map = action_map.get(env_name)

     agent_type, env_name = props[0], props[1]
+    agent = AGENTS_MAP[agent_type](env=env_name, render_mode="rgb_array")
     agent.load_policy(policy_path)
     env_action_map = action_map.get(env_name)