From 0caf33fa9b6eddc71aa551efa574032669414b2a Mon Sep 17 00:00:00 2001 From: Bihan Rana Date: Mon, 26 Feb 2024 22:53:15 +0545 Subject: [PATCH] Remove a location from test_locations integrity test --- src/gpuhunt/providers/cudo.py | 2 +- src/integrity_tests/test_cudo.py | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/src/gpuhunt/providers/cudo.py b/src/gpuhunt/providers/cudo.py index fd1272f..33da4d1 100644 --- 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 index 5c282d1..6dd3062 100644 --- 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)