aws_communication_success

desktop_env/desktop_env.py

@@ -23,10 +23,10 @@ class DesktopEnv(gym.Env):
     """
     DesktopEnv with OpenAI Gym interface. It provides a desktop environment for setting and evaluating desktop automation tasks.
     """
-
+    #TODO:provider_name: str = "vmware",
     def __init__(
             self,
-            provider_name: str = "vmware",
+            provider_name: str = "aws",
             region: str = None,
             path_to_vm: str = None,
             snapshot_name: str = "init_state",
@@ -55,7 +55,7 @@ class DesktopEnv(gym.Env):
         # Initialize VM manager and vitualization provider
         self.region = region
 
-        # Default
+        # Default TODO:
         self.server_port = 5000
         self.chromium_port = 9222
         self.vnc_port = 8006
@@ -69,7 +69,7 @@ class DesktopEnv(gym.Env):
             self.path_to_vm = os.path.abspath(os.path.expandvars(os.path.expanduser(path_to_vm))) \
                 if provider_name in {"vmware", "virtualbox"} else path_to_vm
         else:
-            self.path_to_vm = self.manager.get_vm_path(self.os_type, region)
+            self.path_to_vm = self.manager.get_vm_path(region) # self.os_type,
 
         self.snapshot_name = snapshot_name
         self.cache_dir_base: str = cache_dir
@@ -114,7 +114,8 @@ class DesktopEnv(gym.Env):
         # due to the fact it could be changed when implemented by cloud services
         path_to_vm = self.provider.revert_to_snapshot(self.path_to_vm, self.snapshot_name)
         if path_to_vm and not path_to_vm == self.path_to_vm:
-            # path_to_vm has to be a new path
+            # path_to_vm has to be a new path 
+            
             self.manager.delete_vm(self.path_to_vm, self.region)
             self.manager.add_vm(path_to_vm, self.region)
             self.manager.occupy_vm(path_to_vm, os.getpid(), self.region)