added ogail script and splitting up feasability on next line to avoid calculating it unnecessarily (though it might be fine as is)

2022-02-25 18:39:33 -08:00
parent fa98601fa6
commit c28c6c05b7
2 changed files with 177 additions and 3 deletions
--- a/src/safe_options/options.py
+++ b/src/safe_options/options.py
@@ -249,9 +249,9 @@ class SafeOptionsEnv(OptionsEnv):
            if d:
                break

-            if self.abort_unsafe_collision_method is not None and \
-                not feasible(self.env, plan[k:], method=self.abort_unsafe_collision_method):
-                break
+            if self.abort_unsafe_collision_method is not None:
+                if not feasible(self.env, plan[k:], method=self.abort_unsafe_collision_method):
+                    break
        
        n_steps = k + 1
        return observations, actions, rewards, env_done, plan_done, infos, n_steps