ark-shopify commited on
Commit
d5c1be1
·
unverified ·
1 Parent(s): 05fc91d

fix: Launchers - HuggingFace - Fixed handling of the new SCHEDULING Job status

Browse files

HF uses the "SCHEDULING" status, but it's missing from the client library: https://github.com/huggingface/huggingface_hub/blame/50013bdfb6879fb49c94cc7ade5b8c10f59000c0/src/huggingface_hub/_jobs_api.py#L32

huggingface_overlay/cloud_pipelines_backend/launchers/huggingface_launchers.py CHANGED
@@ -358,6 +358,9 @@ class LaunchedHuggingFaceJobContainer(interfaces.LaunchedContainer):
358
  return interfaces.ContainerStatus.FAILED
359
  elif status_str == huggingface_hub.JobStage.CANCELED:
360
  return interfaces.ContainerStatus.FAILED
 
 
 
361
  else: # "DELETED"
362
  return interfaces.ContainerStatus.ERROR
363
 
 
358
  return interfaces.ContainerStatus.FAILED
359
  elif status_str == huggingface_hub.JobStage.CANCELED:
360
  return interfaces.ContainerStatus.FAILED
361
+ # HF uses the "SCHEDULING" status, but it's missing from the client library: https://github.com/huggingface/huggingface_hub/blame/50013bdfb6879fb49c94cc7ade5b8c10f59000c0/src/huggingface_hub/_jobs_api.py#L32
362
+ elif status_str == "SCHEDULING":
363
+ return interfaces.ContainerStatus.PENDING
364
  else: # "DELETED"
365
  return interfaces.ContainerStatus.ERROR
366