Support AWS G7e (RTXPRO6000) instances (#227)

Author: jvstme

src/gpuhunt/providers/aws.py

@@ -84,6 +84,9 @@
     "ca-west-1",
     "me-south-1",
 }
+GPU_NAME_MAPPING = {
+    "RTX PRO Server 6000": "RTXPRO6000",
+}
 
 
 class AWSProvider(AbstractProvider):
@@ -184,7 +187,7 @@ def fill_gpu_details(self, offers: list[RawCatalogItem]):
                             if "GpuInfo" in i:
                                 gpu = i["GpuInfo"]["Gpus"][0]
                                 gpus[i["InstanceType"]] = (
-                                    gpu["Name"],
+                                    GPU_NAME_MAPPING.get(gpu["Name"], gpu["Name"]),
                                     _get_gpu_memory_gib(
                                         gpu["Name"], gpu["MemoryInfo"]["SizeInMiB"]
                                     ),
@@ -379,6 +382,7 @@ def filter(cls, offers: list[RawCatalogItem]) -> list[RawCatalogItem]:
                     "p4d.",
                     "p4de.",
                     "p3.",
+                    "g7e.",
                     "g6.",
                     "g6e.",
                     "gr6.",

src/integrity_tests/test_aws.py

@@ -60,6 +60,7 @@ def test_gpu_presented(self, data: str):
             "T4",
             "V100",
             "L40S",
+            "RTXPRO6000",
             "L4",
         ]
         assert all(f",{i}," in data for i in gpus)