Skip to content

Commit

Permalink
Revert "Remove "se-smedjebacken-1" from cudo locations"
Browse files Browse the repository at this point in the history
This reverts commit a85da21.
  • Loading branch information
Sergey Mezentsev authored and TheBits committed Mar 1, 2024
1 parent 64e0deb commit 275446b
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 9 deletions.
2 changes: 1 addition & 1 deletion .pre-commit-config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -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
9 changes: 1 addition & 8 deletions src/gpuhunt/providers/cudo.py
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand Down Expand Up @@ -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,
Expand Down
1 change: 1 addition & 0 deletions src/integrity_tests/test_cudo.py
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down

0 comments on commit 275446b

Please sign in to comment.