Fix cudo integrity test import

dstackai · Feb 26, 2024 · de76f1b · de76f1b
1 parent 2496090
commit de76f1b
Showing 1 changed file with 9 additions and 11 deletions.
diff --git a/src/integrity_tests/test_cudo.py b/src/integrity_tests/test_cudo.py
@@ -5,7 +5,7 @@
 
 import pytest
 
-from src.gpuhunt.providers.cudo import GPU_MAP
+from gpuhunt.providers.cudo import GPU_MAP
 
 
 @pytest.fixture
@@ -21,16 +21,14 @@ def select_row(rows, name: str) -> List[str]:
 
 
 def test_locations(data_rows):
-    expected = set(
-        (
-            "no-luster-1",
-            "se-smedjebacken-1",
-            "gb-london-1",
-            "se-stockholm-1",
-            "us-newyork-1",
-            "us-santaclara-1",
-        )
-    )
+    expected = {
+        "no-luster-1",
+        "se-smedjebacken-1",
+        "gb-london-1",
+        "se-stockholm-1",
+        "us-newyork-1",
+        "us-santaclara-1",
+    }
     locations = select_row(data_rows, "location")
     assert set(locations) == expected