diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index edede9f..84fe3db 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -12,6 +12,6 @@ repos: name: isort (python) args: ['--settings-file', 'pyconfig.toml'] - repo: https://github.com/astral-sh/ruff-pre-commit - rev: v0.3.0 + rev: v0.2.2 hooks: - id: ruff diff --git a/src/gpuhunt/providers/cudo.py b/src/gpuhunt/providers/cudo.py index 357fb35..9a7da10 100644 --- a/src/gpuhunt/providers/cudo.py +++ b/src/gpuhunt/providers/cudo.py @@ -15,10 +15,6 @@ API_URL = "https://rest.compute.cudo.org/v1" -SKIP_DC = [ - "se-smedjebacken-1", # sometimes there are no available VMs -] - class CudoProvider(AbstractProvider): NAME = "cudo" @@ -56,12 +52,9 @@ def get_raw_catalog_list(self, vm_machine_type_list, vcpu, memory, gpu: int): if gpu and name is None: logger.warning("Skip. Unknown GPU name: %s", vm["gpuModel"]) continue - data_center = vm["dataCenterId"] - if data_center in SKIP_DC: - continue raw = RawCatalogItem( instance_name=vm["machineType"], - location=data_center, + location=vm["dataCenterId"], spot=False, price=round(float(vm["totalPriceHr"]["value"]), 5), cpu=vcpu, diff --git a/src/integrity_tests/test_cudo.py b/src/integrity_tests/test_cudo.py index 5efe35c..c23e92c 100644 --- a/src/integrity_tests/test_cudo.py +++ b/src/integrity_tests/test_cudo.py @@ -24,6 +24,7 @@ def select_row(rows, name: str) -> List[str]: def test_locations(data_rows): expected = { "no-luster-1", + "se-smedjebacken-1", "se-stockholm-1", "us-newyork-1", "us-santaclara-1",