diff --git a/eval_experiments.py b/eval_experiments.py index 3eed2ed..f9f1c8e 100644 --- a/eval_experiments.py +++ b/eval_experiments.py @@ -12,7 +12,7 @@ def main(method:str='expert', folder:str=None, locations=[(0,0)], skip_running=F env_kwargs={'stop_on_collision':True, 'max_episode_steps':1000} elif method in ['hail']: env = 'NormalizedOptionsEvalEnv' - env_kwargs={'stop_on_collision':True, 'max_episode_steps':1000} + env_kwargs={'stop_on_collision':True, 'max_episode_steps':1000, 'safe_actions_collision_method': None, 'abort_unsafe_collision_method': None} elif method in ['shail']: env = 'NormalizedSafeOptionsEvalEnv' env_kwargs={'stop_on_collision':True, 'max_episode_steps':1000} diff --git a/evaluate_models.sh b/evaluate_models.sh index 85c0add..8ec9a15 100755 --- a/evaluate_models.sh +++ b/evaluate_models.sh @@ -8,8 +8,8 @@ python -m eval_experiments --method bc --folder='test_policies/bc/expA' python -m eval_experiments --method bc --folder='test_policies/bc/expB' --locations='[(0,4)]' python -m eval_experiments --method gail --folder='test_policies/gail/expA' python -m eval_experiments --method gail --folder='test_policies/gail/expB' --locations='[(0,4)]' -python -m eval_experiments --method hail --folder='test_policies/hail/expA' --env_kwargs='{safe_actions_collision_method=None,abort_unsafe_collision_method=None}' -python -m eval_experiments --method hail --folder='test_policies/hail/expB' --locations='[(0,4)]' --env_kwargs='{safe_actions_collision_method=None,abort_unsafe_collision_method=None}' +python -m eval_experiments --method hail --folder='test_policies/hail/expA' +python -m eval_experiments --method hail --folder='test_policies/hail/expB' --locations='[(0,4)]' python -m eval_experiments --method shail --folder='test_policies/shail/expA' python -m eval_experiments --method shail --folder='test_policies/shail/expB' --locations='[(0,4)]'