Remove a location from test_locations integrity test

dstackai · Feb 26, 2024 · 0caf33f · 0caf33f
1 parent 85d95a2
commit 0caf33f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/src/gpuhunt/providers/cudo.py b/src/gpuhunt/providers/cudo.py
@@ -53,7 +53,7 @@ def get_raw_catalog_list(self, vm_machine_type_list, vcpu, memory, gpu):
                 cpu=vcpu,
                 memory=memory,
                 gpu_count=gpu,
-                gpu_name=vm["gpuModel"],
+                gpu_name=gpu_name(vm["gpuModel"]),
                 gpu_memory=get_memory(gpu_name(vm["gpuModel"])),
                 disk_size=None,
             )

diff --git a/src/integrity_tests/test_cudo.py b/src/integrity_tests/test_cudo.py
@@ -24,7 +24,6 @@ def test_locations(data_rows):
     expected = {
         "no-luster-1",
         "se-smedjebacken-1",
-        "gb-london-1",
         "se-stockholm-1",
         "us-newyork-1",
         "us-santaclara-1",
@@ -45,4 +44,4 @@ def test_price(data_rows):
 def test_gpu_present(data_rows):
     refs = GPU_MAP.values()
     gpus = select_row(data_rows, "gpu_name")
-    assert all(i in gpus for i in refs)
+    assert all(i in refs for i in gpus)