fix a small bug in computer_13 action space

2024-06-11 14:22:31 +08:00
parent 1c50770817
commit c121869219
2 changed files with 2 additions and 2 deletions
--- a/desktop_env/controllers/python.py
+++ b/desktop_env/controllers/python.py
@@ -144,7 +144,7 @@ class PythonController:
            return

        action_type = action["action_type"]
-        parameters = action["parameters"] if "parameters" in action else {}
+        parameters = action["parameters"] if "parameters" in action else {param: action[param] for param in action if param != 'action_type'}
        move_mode = random.choice(
            ["pyautogui.easeInQuad", "pyautogui.easeOutQuad", "pyautogui.easeInOutQuad", "pyautogui.easeInBounce",
             "pyautogui.easeInElastic"])