feat: transition to new runtime names

Author: mpetazzoni

README.md

@@ -33,7 +33,7 @@ from wherobots.db.runtime import Runtime
 
 with connect(
         api_key='...',
-        runtime=Runtime.SEDONA,
+        runtime=Runtime.TINY,
         region=Region.AWS_US_WEST_2) as conn:
     curr = conn.cursor()
     curr.execute("SHOW SCHEMAS IN wherobots_open_data")

tests/smoke.py

@@ -71,8 +71,6 @@
             host=args.api_endpoint,
             token=token,
             api_key=api_key,
-            runtime=Runtime.SEDONA,
-            region=Region.AWS_US_WEST_2,
             shutdown_after_inactive_seconds=args.shutdown_after_inactive_seconds,
             wait_timeout=900,
         )

wherobots/db/constants.py

@@ -7,7 +7,7 @@
 
 DEFAULT_ENDPOINT: str = "api.cloud.wherobots.com"  # "api.cloud.wherobots.com"
 STAGING_ENDPOINT: str = "api.staging.wherobots.com"  # "api.staging.wherobots.com"
-DEFAULT_RUNTIME: Runtime = Runtime.SEDONA
+DEFAULT_RUNTIME: Runtime = Runtime.TINY
 DEFAULT_REGION: Region = Region.AWS_US_WEST_2
 DEFAULT_READ_TIMEOUT_SECONDS: float = 0.25
 DEFAULT_SESSION_WAIT_TIMEOUT_SECONDS: float = 900

wherobots/db/runtime.py

@@ -2,6 +2,27 @@
 
 
 class Runtime(Enum):
+    TINY = "tiny"
+    SMALL = "small"
+    MEDIUM = "medium"
+    LARGE = "large"
+    X_LARGE = "x-large"
+    XX_LARGE = "2x-large"
+    XXXX_LARGE = "4x-large"
+
+    # HIMEM
+    MEDIUM_HIMEM = "medium-himem"
+    LARGE_HIMEM = "large-himem"
+    X_LARGE_HIMEM = "x-large-himem"
+    XX_LARGE_HIMEM = "2x-large-himem"
+    XXXX_LARGE_HIMEM = "4x-large-himem"
+
+    # GPU
+    TINY_A10_GPU = "tiny-a10-gpu"
+    SMALL_A10_GPU = "small-a10-gpu"
+    MEDIUM_A10_GPU = "medium-a10-gpu"
+
+    # Deprecated names; will be removed in a later major version.
     SEDONA = "tiny"
     SAN_FRANCISCO = "small"
     NEW_YORK = "medium"