Fix cudo integrity tests (#40)

* Add CudoCompute provider Added cudocompute in OFFLINE PROVIDERS Removed dist directory Cudocompute disksize set to None * Fixed Review Issues and Added Integrity Tests for Cudo * Adjust Spacing With Pre-Commit for Cudo * Remove cudo-compute python client * Remove a location from test_locations integrity test * Re-Added Install Dependencies for Cudo --------- Co-authored-by: Bihan Rana <[email protected]>
dstackai · Feb 27, 2024 · 1e4be3d · 1e4be3d
1 parent 7e4c7d8
commit 1e4be3d
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)