From c4e0d49c1c31dcfc0f2bc5cfbaa17eb65d95578d Mon Sep 17 00:00:00 2001 From: peterschmidt85 Date: Tue, 26 Aug 2025 10:42:03 +0200 Subject: [PATCH] Fix regarding https://github.com/dstackai/dstack/issues/2962 --- src/gpuhunt/providers/nebius.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/src/gpuhunt/providers/nebius.py b/src/gpuhunt/providers/nebius.py index ce3ba3d..c7c1245 100644 --- a/src/gpuhunt/providers/nebius.py +++ b/src/gpuhunt/providers/nebius.py @@ -165,12 +165,14 @@ def get_regions_map(sdk: SDK) -> dict[str, str]: Returns: `{"e00": "eu-north1", "e01": "eu-west1", ...}` """ - tenants = TenantServiceClient(sdk).list(ListTenantsRequest(), timeout=TIMEOUT).wait() + tenants = TenantServiceClient(sdk).list(ListTenantsRequest(), per_retry_timeout=TIMEOUT).wait() if len(tenants.items) != 1: raise ValueError(f"Expected to find 1 tenant, found {(len(tenants.items))}") projects = ( ProjectServiceClient(sdk) - .list(ListProjectsRequest(parent_id=tenants.items[0].metadata.id), timeout=TIMEOUT) + .list( + ListProjectsRequest(parent_id=tenants.items[0].metadata.id), per_retry_timeout=TIMEOUT + ) .wait() ) result = {} @@ -188,7 +190,7 @@ def list_platforms(sdk: SDK, region_code: str) -> ListPlatformsResponse: page_size=999, parent_id=f"project-{region_code}public-images", ) - return PlatformServiceClient(sdk).list(req, timeout=TIMEOUT).wait() + return PlatformServiceClient(sdk).list(req, per_retry_timeout=TIMEOUT).wait() def make_item(