From 938a3bb918a3d1bca6075f94280256b230c69b8d Mon Sep 17 00:00:00 2001
From: Timothyxxx <384084775@qq.com>
Date: Thu, 22 Feb 2024 20:39:07 +0800
Subject: [PATCH] Fix get IP error

---
 desktop_env/envs/desktop_env.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/desktop_env/envs/desktop_env.py b/desktop_env/envs/desktop_env.py
index ed2f8c4..fbfb739 100644
--- a/desktop_env/envs/desktop_env.py
+++ b/desktop_env/envs/desktop_env.py
@@ -137,7 +137,7 @@ class DesktopEnv(gym.Env):
         logger.info("Getting IP Address...")
         for _ in range(max_retries):
             try:
-                output = _execute_command(["vmrun", "-T", "ws", "getGuestIPAddress", self.path_to_vm]).strip()
+                output = _execute_command(["vmrun", "-T", "ws", "getGuestIPAddress", self.path_to_vm, "-wait"]).strip()
                 logger.info(f"IP address: {output}")
                 return output
             except Exception as e: