diff --git a/sky/clouds/service_catalog/data_fetchers/fetch_fluidstack.py b/sky/clouds/service_catalog/data_fetchers/fetch_fluidstack.py index 7a8b7e42e79..f56016f8317 100644 --- a/sky/clouds/service_catalog/data_fetchers/fetch_fluidstack.py +++ b/sky/clouds/service_catalog/data_fetchers/fetch_fluidstack.py @@ -168,8 +168,8 @@ GPU_MAP = { 'H100_PCIE_80GB': 'H100', - 'H100_NVLINK_80GB': 'H100', - 'A100_NVLINK_80GB': 'A100-80GB', + 'H100_NVLINK_80GB': 'H100-NVLINK', + 'A100_NVLINK_80GB': 'A100-80GB-NVLINK', 'A100_SXM4_80GB': 'A100-80GB-SXM', 'H100_SXM5_80GB': 'H100-SXM', 'A100_PCIE_80GB': 'A100-80GB', @@ -206,6 +206,8 @@ def create_catalog(output_dir: str) -> None: with open(DEFAULT_FLUIDSTACK_API_KEY_PATH, 'r', encoding='UTF-8') as f: api_key = f.read().strip() response = requests.get(ENDPOINT, headers={'api-key': api_key}) + if not response.ok: + raise Exception(response.text) plans = response.json() with open(os.path.join(output_dir, 'vms.csv'), mode='w', diff --git a/sky/provision/fluidstack/instance.py b/sky/provision/fluidstack/instance.py index 1bb123ef649..9860b8a8345 100644 --- a/sky/provision/fluidstack/instance.py +++ b/sky/provision/fluidstack/instance.py @@ -298,7 +298,7 @@ def query_instances( 'pending': status_lib.ClusterStatus.INIT, 'stopped': status_lib.ClusterStatus.STOPPED, 'running': status_lib.ClusterStatus.UP, - 'unhealthy': status_lib.ClusterStatus.INIT, + 'failed': status_lib.ClusterStatus.INIT, 'terminated': None, } statuses: Dict[str, Optional[status_lib.ClusterStatus]] = {}