Fix benchmark script

araffin · Mar 15, 2020 · 7cb227c · 7cb227c
1 parent 5ab97c8
commit 7cb227c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/benchmark.py b/utils/benchmark.py
@@ -56,7 +56,7 @@
         continue
 
     # Skip old BipedalWalker version
-    if 'Walker-v2' in trained_model or 'WalkerHardcore-v2' in trained_model:
+    if 'Walker-v2' in env_id or 'WalkerHardcore-v2' in env_id:
         continue
 
     reward_log = '{}/{}/'.format(args.benchmark_dir, trained_model)