Skip to content

Commit dddfe09

Browse files
kaiisfreeKaiclaudekyuds
authored
Fix PrimeIntellect query_instances TypeError on retry_if_missing (#9259)
* Fix PrimeIntellect query_instances missing retry_if_missing parameter The PrimeIntellect provider's query_instances() was missing the retry_if_missing parameter that the base interface expects, causing a TypeError when SkyPilot tried to query cluster status. Fixes #9256 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> * Address review: add missing cluster_name param and PROVISIONING status mapping Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> * Update sky/provision/primeintellect/instance.py --------- Co-authored-by: Kai <kai@kaiisfree.com> Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com> Co-authored-by: Daniel Shin <88547237+kyuds@users.noreply.github.com>
1 parent 290fe13 commit dddfe09

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

sky/provision/primeintellect/instance.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -421,16 +421,20 @@ def get_cluster_info(
421421

422422

423423
def query_instances(
424+
cluster_name: str,
424425
cluster_name_on_cloud: str,
425426
provider_config: Optional[Dict[str, Any]] = None,
426427
non_terminated_only: bool = True,
428+
retry_if_missing: bool = False,
427429
) -> Dict[str, Tuple[Optional['status_lib.ClusterStatus'], Optional[str]]]:
428430
"""See sky/provision/__init__.py"""
431+
del cluster_name, retry_if_missing # unused
429432
assert provider_config is not None, (cluster_name_on_cloud, provider_config)
430433
instances = _filter_instances(cluster_name_on_cloud, None)
431434

432435
status_map = {
433436
'PENDING': status_lib.ClusterStatus.INIT,
437+
'PROVISIONING': status_lib.ClusterStatus.INIT,
434438
'ERROR': status_lib.ClusterStatus.INIT,
435439
'ACTIVE': status_lib.ClusterStatus.UP,
436440
'STOPPED': status_lib.ClusterStatus.STOPPED,

0 commit comments

Comments
 (0)