root
/
OSWorld
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

ver Feb2ndv3 

working on human eval for multi_apps
This commit is contained in:
David Chang 2024-02-02 09:30:10 +08:00
parent 6d792e594e
commit c46fcbfcbe
No known key found for this signature in database
GPG Key ID: B5C4E1C04C58CAA7
2 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
desktop_env/controllers/setup.py
View File

@ -406,7 +406,7 @@ class SetupController:
except Exception as e:
if attempt < 14:
logger.error(f"Attempt {attempt + 1}: Failed to connect, retrying. Error: {e}")
time.sleep(1)
time.sleep(5)
else:
logger.error(f"Failed to connect after multiple attempts: {e}")
raise e
@ -531,20 +531,23 @@ class SetupController:
except Exception as e:
if attempt < 14:
logger.error(f"Attempt {attempt + 1}: Failed to connect, retrying. Error: {e}")
time.sleep(1)
time.sleep(5)
else:
logger.error(f"Failed to connect after multiple attempts: {e}")
raise e
if not browser:
return
context = browser.contexts[0]
platform = config['platform']
if platform == 'googledrive':
url = 'https://drive.google.com/drive/my-drive'
page = context.new_page() # Create a new page (tab) within the existing context
page.goto(url)
try:
page.goto(url, timeout=60000)
except:
logger.warning("Opening %s exceeds time limit", url) # only for human test
logger.info(f"Opened new page: {url}")
settings = json.load(open(config['settings_file']))
email, password = settings['email'], settings['password']

4
main.py
View File

@ -47,9 +47,9 @@ def human_agent():
Runs the Gym environment with human input.
"""
with open("evaluation_examples/examples/vlc/215dfd39-f493-4bc3-a027-8a97d72c61bf.json", "r") as f:
with open("evaluation_examples/examples/multi_apps/510f64c8-9bcc-4be1-8d30-638705850618.json", "r") as f:
example = json.load(f)
example["snapshot"] = "Snapshot 36"
example["snapshot"] = "Snapshot 35"
#env = DesktopEnv( path_to_vm="~/vmware/Windows 10 x64/Windows 10 x64.vmx"
env = DesktopEnv( path_to_vm="/mnt/data1/david/os-images/Ubuntu-1218/Ubuntu.vmx"