fix demos evaluation

JunhaoWang · Feb 19, 2019 · 8964a6d · 8964a6d
1 parent a2db811
commit 8964a6d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/scripts/evaluate.py b/scripts/evaluate.py
@@ -97,7 +97,8 @@ def main(args, seed, episodes):
     if n > 0:
         print("{} worst episodes:".format(n))
         for i in indexes[:n]:
-            print(logs['seed_per_episode'][i])
+            if 'seed_per_episode' in logs:
+                print(logs['seed_per_episode'][i])
             if args.model is not None:
                 print("- episode {}: R={}, F={}".format(i, logs["return_per_episode"][i], logs["num_frames_per_episode"][i]))
             else: