From 5fb358d725ff1e8863d0d4e2119f04686da9bb5a Mon Sep 17 00:00:00 2001
From: ebuehrle <43623224+ebuehrle@users.noreply.github.com>
Date: Thu, 22 Sep 2022 11:15:49 +0200
Subject: [PATCH] do not enable IDM for expert data

---
 eval_experiments.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/eval_experiments.py b/eval_experiments.py
index e10d346..0c58e7f 100644
--- a/eval_experiments.py
+++ b/eval_experiments.py
@@ -11,8 +11,10 @@ def main(method:str='expert', folder:str=None, locations=[(0,0)], skip_running=F
     exclude_keys_from_policy_kwargs = {'learning_rate', 'learning_rate_decay', 'clip_ratio', 'iterations_per_epoch', 'option'}
     policy_kwargs = {}
 
-    if method in ['expert', 'expert_agent', 'idm']:
+    if method in ['expert', 'expert_agent']:
         env, env_kwargs ='NRasterizedRouteIncrementingAgent', {}
+    elif method in ['idm']:
+        env, env_kwargs ='NRasterizedRouteIncrementingAgent', {'use_idm':True}
     elif method in ['bc','gail']:
         env='NormalizedContinuousEvalEnv' 
         env_kwargs={'stop_on_collision':True, 'max_episode_steps':1000, 'use_idm':True}