DataCrunch: Add the FIN-02 datacenter

dstackai · Mar 12, 2024 · 6096564 · 6096564
1 parent 275446b
commit 6096564
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -46,7 +46,7 @@ List of all available filters:
 from gpuhunt import Catalog
 
 catalog = Catalog()
-catalog.load(version="20230831")
+catalog.load(version="20240310")
 items = catalog.query()
 
 print(*items, sep="\n")

diff --git a/src/integrity_tests/test_datacrunch.py b/src/integrity_tests/test_datacrunch.py
@@ -20,7 +20,11 @@ def select_row(rows, name: str) -> List[str]:
 
 
 def test_locations(data_rows):
-    expected = set(("FIN-01", "ICE-01"))
+    expected = {
+        "FIN-01",
+        "FIN-02",
+        "ICE-01",
+    }
     locations = select_row(data_rows, "location")
     assert set(locations) == expected
 

diff --git a/src/tests/providers/test_datacrunch.py b/src/tests/providers/test_datacrunch.py
@@ -147,6 +147,7 @@ def locations():
     # datacrunch.locations.get()
     return [
         {"code": "FIN-01", "name": "Finland 1", "country_code": "FI"},
+        {"code": "FIN-02", "name": "Finland 2", "country_code": "FI"},
         {"code": "ICE-01", "name": "Iceland 1", "country_code": "IS"},
     ]