feat: use SSD with high throughput (#218)

desktop_env/providers/aws/manager.py

@@ -6,7 +6,7 @@ import logging
 import dotenv
 import signal
 
-INSTANCE_TYPE = "t3.large"
+INSTANCE_TYPE = "t3.medium" 
 
 # Load environment variables from .env file
 dotenv.load_dotenv()
@@ -97,6 +97,15 @@ def _allocate_vm(region=DEFAULT_REGION):
                         os.getenv('AWS_SECURITY_GROUP_ID')
                     ]
                 }
+            ],
+            "BlockDeviceMappings": [
+                {
+                    "DeviceName": "/dev/sda1", 
+                    "Ebs": {
+                        'VolumeType': 'io2',
+                        'Iops': 16000,
+                    }
+                }
             ]
         }
         

desktop_env/providers/aws/provider.py

@@ -128,6 +128,15 @@ class AWSProvider(Provider):
                         "DeviceIndex": 0,
                         "Groups": security_groups
                     }
+                ],
+                "BlockDeviceMappings": [
+                    {
+                        "DeviceName": "/dev/sda1", 
+                        "Ebs": {
+                            'VolumeType': 'io2',
+                            'Iops': 16000,
+                        }
+                    }
                 ]
             }