Fix action_space setup

2023-12-06 22:59:19 +08:00
parent b9c317f0f5
commit 343b40ecac
3 changed files with 68 additions and 236 deletions
--- a/desktop_env/controllers/setup.py
+++ b/desktop_env/controllers/setup.py
@@ -21,6 +21,8 @@ class SetupController:
        # can add other setup steps

    def _download_setup(self, config):
+        if not config:
+            return
        if not 'download' in config:
            return
        for url, path in config['download']:
@@ -43,6 +45,8 @@ class SetupController:
                print("An error occurred while trying to send the request:", e)

    def _change_wallpaper(self, config):
+        if not config:
+            return
        if not 'wallpaper' in config:
            return
        path = config['wallpaper']
@@ -68,6 +72,8 @@ class SetupController:
        raise NotImplementedError

    def _open_setup(self, config):
+        if not config:
+            return
        if not 'open' in config:
            return
        for path in config['open']: