Update on Chrome examples; Refactor on logic of controlling

2024-01-12 17:24:47 +08:00
parent 820579a5a2
commit 5a93a32958
17 changed files with 575 additions and 194 deletions
--- a/desktop_env/envs/desktop_env.py
+++ b/desktop_env/envs/desktop_env.py
@@ -80,9 +80,8 @@ class DesktopEnv(gym.Env):
        logger.info("Initializing...")
        self._start_emulator()
        self.vm_ip = self._get_vm_ip()
-        self.host = f"http://{self.vm_ip}:5000"
-        self.controller = PythonController(http_server=self.host)
-        self.setup_controller = SetupController(http_server=self.host, cache_dir=self.cache_dir)
+        self.controller = PythonController(vm_ip=self.vm_ip)
+        self.setup_controller = SetupController(vm_ip=self.vm_ip, cache_dir=self.cache_dir)

        # Meta info of the VM
        self.vm_platform = self.controller.get_vm_platform()