Change cudo to online provider

dstackai · Mar 12, 2024 · 92bf37c · 92bf37c
1 parent 5efb46d
commit 92bf37c
Show file tree

Hide file tree

Showing 5 changed files with 355 additions and 477 deletions.
diff --git a/src/gpuhunt/_internal/catalog.py b/src/gpuhunt/_internal/catalog.py
@@ -17,8 +17,8 @@
 logger = logging.getLogger(__name__)
 version_url = "https://dstack-gpu-pricing.s3.eu-west-1.amazonaws.com/v1/version"
 catalog_url = "https://dstack-gpu-pricing.s3.eu-west-1.amazonaws.com/v1/{version}/catalog.zip"
-OFFLINE_PROVIDERS = ["aws", "azure", "cudo", "datacrunch", "gcp", "lambdalabs", "nebius"]
-ONLINE_PROVIDERS = ["tensordock", "vastai"]
+OFFLINE_PROVIDERS = ["aws", "azure", "datacrunch", "gcp", "lambdalabs", "nebius"]
+ONLINE_PROVIDERS = ["cudo", "tensordock", "vastai"]
 RELOAD_INTERVAL = 4 * 60 * 60  # 4 hours
 
 

diff --git a/src/gpuhunt/_internal/default.py b/src/gpuhunt/_internal/default.py
@@ -21,6 +21,7 @@ def default_catalog() -> Catalog:
     for module, provider in [
         ("gpuhunt.providers.tensordock", "TensorDockProvider"),
         ("gpuhunt.providers.vastai", "VastAIProvider"),
+        ("gpuhunt.providers.cudo", "CudoProvider")
     ]:
         try:
             module = importlib.import_module(module)