Improve on eval script on web browsing tasks; Add one setup example

2024-02-23 11:57:50 +08:00
parent 543666ff4d
commit 81863b26dd
4 changed files with 72 additions and 41 deletions
--- a/desktop_env/envs/desktop_env.py
+++ b/desktop_env/envs/desktop_env.py
@@ -316,6 +316,9 @@ class DesktopEnv(gym.Env):
                return 1
            else:
                return 0
+        else:
+            if len(self.action_history) > 0 and self.action_history[-1] == "FAIL":
+                return 0

        if type(self.metric) == list:
            results = []