diff --git a/desktop_env/envs/desktop_env.py b/desktop_env/envs/desktop_env.py index d7b1b98..bbf0620 100644 --- a/desktop_env/envs/desktop_env.py +++ b/desktop_env/envs/desktop_env.py @@ -91,7 +91,6 @@ class DesktopEnv(gym.Env): return mouse_controller, keyboard_controller def _start_emulator(self): - # fixme: check if the vm is running while True: try: output = subprocess.check_output(f"vmrun -T ws list", shell=True, stderr=subprocess.STDOUT) @@ -113,8 +112,6 @@ class DesktopEnv(gym.Env): self._execute_command(["vmrun", "-T", "ws" "snapshot", self.path_to_vm, self.snapshot_path]) def _get_screenshot(self): - # todo: hash it and store it in a temporary directory - random_uuid = str(uuid.uuid4()) os.makedirs(os.path.join("tmp", random_uuid), exist_ok=True) image_path = os.path.join("tmp", random_uuid, "screenshot.png")