o j8@sddlmZddlmZddlmZddlmZddlmZddl m Z m Z ddl m Z Gdd d eeZeGd d d ZeGd d d ZeGdddZeGdddZeGdddZeGdddZeGdddZeGdddZeGdddZddddedeed eeefdBd!eeefdBd"e dBd#eeBeBdBd$eeefdBd%ee dBd&eeeffd'd(ZdS))) dataclass)datetime)Enum)Any) constants) SpaceHardwareVolume)parse_datetimec@s$eZdZdZdZdZdZdZdZdS)JobStageah Enumeration of possible stage of a Job on the Hub. Value can be compared to a string: ```py assert JobStage.COMPLETED == "COMPLETED" ``` Possible values are: `COMPLETED`, `CANCELED`, `ERROR`, `DELETED`, `RUNNING`. Taken from https://github.com/huggingface/moon-landing/blob/main/server/job_types/JobInfo.ts#L61 (private url). COMPLETEDCANCELEDERRORDELETEDRUNNINGN) __name__ __module__ __qualname____doc__r r r rrrrL/home/kuhnn/.local/lib/python3.10/site-packages/huggingface_hub/_jobs_api.pyr s r c@s"eZdZUeed<edBed<dS) JobStatusstageNmessage)rrrr __annotations__strrrrrr,s rc@s&eZdZUeed<eed<eed<dS)JobOwneridnametypeN)rrrrrrrrrr2s  rc@seZdZUdZeed<edBed<edBed<edBed<eedBed<eedBed<eee fdBed <eee fdBed <e dBed <eeefdBed <ee dBed <e ed<e ed<eed<eed<dddZdS)JobInfoa Contains information about a Job. Args: id (`str`): Job ID. created_at (`datetime` or `None`): When the Job was created. docker_image (`str` or `None`): The Docker image from Docker Hub used for the Job. Can be None if space_id is present instead. space_id (`str` or `None`): The Docker image from Hugging Face Spaces used for the Job. Can be None if docker_image is present instead. command (`list[str]` or `None`): Command of the Job, e.g. `["python", "-c", "print('hello world')"]` arguments (`list[str]` or `None`): Arguments passed to the command environment (`dict[str]` or `None`): Environment variables of the Job as a dictionary. secrets (`dict[str]` or `None`): Secret environment variables of the Job (encrypted). flavor (`str` or `None`): Flavor for the hardware, as in Hugging Face Spaces. See [`SpaceHardware`] for possible values. E.g. `"cpu-basic"`. labels (`dict[str, str]` or `None`): Labels to attach to the job (key-value pairs). volumes (`list[Volume]` or `None`): Volumes mounted in the job container (buckets, models, datasets, spaces). status: (`JobStatus` or `None`): Status of the Job, e.g. `JobStatus(stage="RUNNING", message=None)` See [`JobStage`] for possible stage values. owner: (`JobOwner` or `None`): Owner of the Job, e.g. `JobOwner(id="5e9ecfc04957053f60648a3e", name="lhoestq", type="user")` Example: ```python >>> from huggingface_hub import run_job >>> job = run_job( ... image="python:3.12", ... command=["python", "-c", "print('Hello from the cloud!')"] ... ) >>> job JobInfo(id='687fb701029421ae5549d998', created_at=datetime.datetime(2025, 7, 22, 16, 6, 25, 79000, tzinfo=datetime.timezone.utc), docker_image='python:3.12', space_id=None, command=['python', '-c', "print('Hello from the cloud!')"], arguments=[], environment={}, secrets={}, flavor='cpu-basic', labels=None, status=JobStatus(stage='RUNNING', message=None), owner=JobOwner(id='5e9ecfc04957053f60648a3e', name='lhoestq', type='user'), endpoint='https://huggingface.co', url='https://huggingface.co/jobs/lhoestq/687fb701029421ae5549d998') >>> job.id '687fb701029421ae5549d998' >>> job.url 'https://huggingface.co/jobs/lhoestq/687fb701029421ae5549d998' >>> job.status.stage 'RUNNING' ``` rN created_at docker_imagespace_idcommand arguments environmentsecretsflavorlabelsvolumesstatusownerendpointurlreturncKsD|d|_|dp|d}|rt|nd|_|dp!|d|_|dp,|d|_|di}t|d|d |d d |_|d |_|d |_ |d|_ |d|_ |d|_ |d|_ |d}|rtdd|Dnd|_|di}t|d|dd|_|dtj|_|jd|jjd|j|_dS)Nr createdAtr dockerImager!spaceIdr"r+rrrrrr#r$r%r&r'r(r)cSg|] }tdi|qSrr.0vrrr z$JobInfo.__init__..r*rr)rrr,z/jobs//)rgetr r r!r"rr+r#r$r%r&r'r(r)rr*rENDPOINTr,rr-)selfkwargsr r+r)r*rrr__init__s&          "zJobInfo.__init__r.N)rrrrrrrlistdictrrrrrr@rrrrr9s$ 6    rc@seZdZUedBed<edBed<eedBed<eedBed<eeefdBed<eeefdBed<edBed<e dBed <eedBed <edBed <eeefdBed <ee dBed <dddZ dS)JobSpecNr!r"r#r$r%r&r'timeouttagsarchr(r)r.cKs|dp |d|_|dp|d|_|d|_|d|_|d|_|d|_|d |_|d |_|d |_ |d |_ |d |_ |d}|r]dd|D|_ dSd|_ dS)Nr0r!r1r"r#r$r%r&r'rErFrGr(r)cSr3r4r5r6rrrr9r:z$JobSpec.__init__..) r<r!r"r#r$r%r&r'rErFrGr(r))r>r?r)rrrr@s          "zJobSpec.__init__rA) rrrrrrBrCrrintrr@rrrrrDs     rDc@s(eZdZUeed<eed<dddZdS) LastJobInforatr.NcKs|d|_t|d|_dS)NrrJ)rr rJr>r?rrrr@s zLastJobInfo.__init__rA)rrrrrrr@rrrrrIs rIc@s0eZdZUedBed<edBed<dddZdS)ScheduledJobStatusNlast_jobnext_job_run_atr.cKs`|dp |d}|rtdi|nd|_|dp|d}|r+tt||_dSd|_dS)NlastJobrM nextJobRunAtrNr)r<rIrMr rrN)r>r?rMrNrrrr@s zScheduledJobStatus.__init__rA)rrrrIrrr@rrrrrLs  rLc@sleZdZUdZeed<edBed<eed<edBed<edBed<edBed<e ed <e ed <dd d Z dS)ScheduledJobInfoa Contains information about a Job. Args: id (`str`): Scheduled Job ID. created_at (`datetime` or `None`): When the scheduled Job was created. tags (`list[str]` or `None`): The tags of the scheduled Job. schedule (`str` or `None`): One of "@annually", "@yearly", "@monthly", "@weekly", "@daily", "@hourly", or a CRON schedule expression (e.g., '0 9 * * 1' for 9 AM every Monday). suspend (`bool` or `None`): Whether the scheduled job is suspended (paused). concurrency (`bool` or `None`): Whether multiple instances of this Job can run concurrently. status (`ScheduledJobStatus` or `None`): Status of the scheduled Job. owner: (`JobOwner` or `None`): Owner of the scheduled Job, e.g. `JobOwner(id="5e9ecfc04957053f60648a3e", name="lhoestq", type="user")` job_spec: (`JobSpec` or `None`): Specifications of the Job. Example: ```python >>> from huggingface_hub import run_job >>> scheduled_job = create_scheduled_job( ... image="python:3.12", ... command=["python", "-c", "print('Hello from the cloud!')"], ... schedule="@hourly", ... ) >>> scheduled_job.id '687fb701029421ae5549d999' >>> scheduled_job.status.next_job_run_at datetime.datetime(2025, 7, 22, 17, 6, 25, 79000, tzinfo=datetime.timezone.utc) ``` rNr job_specschedulesuspend concurrencyr*r+r.cKs|d|_|dp|d}|rt|nd|_tdi|dp%|di|_|d|_|d|_|d|_|d i}t |d pK|d |d pT|d d|_ |di}t |d|d|dd|_ dS)Nrr/r rRjobSpecrSrTrUr*rMrOrNrP)rMrNr+rrr2r) rr<r r rDrRrSrTrUrLr*rr+)r>r?r r*r+rrrr@s "      zScheduledJobInfo.__init__rA) rrrrrrrrDboolrLrr@rrrrrQs (    rQc@sDeZdZUdZeed<eed<eed<eed<eed<d d d ZdS) JobAcceleratora Contains information about a Job accelerator (GPU). Args: type (`str`): Type of accelerator, e.g. `"gpu"`. model (`str`): Model of accelerator, e.g. `"T4"`, `"A10G"`, `"A100"`, `"L4"`, `"L40S"`. quantity (`str`): Number of accelerators, e.g. `"1"`, `"2"`, `"4"`, `"8"`. vram (`str`): Total VRAM, e.g. `"16 GB"`, `"24 GB"`. manufacturer (`str`): Manufacturer of the accelerator, e.g. `"Nvidia"`. rmodelquantityvram manufacturerr.NcKs6|d|_|d|_|d|_|d|_|d|_dS)NrrYrZr[r\)rrYrZr[r\rKrrrr@,s    zJobAccelerator.__init__rA)rrrrrrr@rrrrrXs rXc@s`eZdZUdZeed<eed<eed<eed<edBed<eed<eed <eed <dd d Z dS) JobHardwarea Contains information about available Job hardware. Args: name (`str`): Machine identifier, e.g. `"cpu-basic"`, `"a10g-large"`. pretty_name (`str`): Human-readable name, e.g. `"CPU Basic"`, `"Nvidia A10G - large"`. cpu (`str`): CPU specification, e.g. `"2 vCPU"`, `"12 vCPU"`. ram (`str`): RAM specification, e.g. `"16 GB"`, `"46 GB"`. accelerator (`JobAccelerator` or `None`): GPU/accelerator details if available. unit_cost_micro_usd (`int`): Cost in micro-dollars per unit, e.g. `167` (= $0.000167). unit_cost_usd (`float`): Cost in USD per unit, e.g. `0.000167`. unit_label (`str`): Cost unit period, e.g. `"minute"`. Example: ```python >>> from huggingface_hub import list_jobs_hardware >>> hardware_list = list_jobs_hardware() >>> hardware_list[0] JobHardware(name='cpu-basic', pretty_name='CPU Basic', cpu='2 vCPU', ram='16 GB', accelerator=None, unit_cost_micro_usd=167, unit_cost_usd=0.000167, unit_label='minute') >>> hardware_list[0].name 'cpu-basic' ``` r pretty_namecpuramN acceleratorunit_cost_micro_usd unit_cost_usd unit_labelr.cKsl|d|_|d|_|d|_|d|_|d}|r"td i|nd|_|d|_|d|_|d|_ dS) Nr prettyNamer_r`raunitCostMicroUSD unitCostUSD unitLabelr) rr^r_r`r<rXrarbrcrd)r>r?rarrrr@`s       zJobHardware.__init__rA) rrrrrrrXrHfloatr@rrrrr]4s ! r]N)r(r)imager#envr&r'rEr(r)r.c Cs|g|pi|p tjd}|r||d<|r?ddddd} t|tr9|d| vr9tt|dd| |d|d <nt||d <|rE||d <|rPd d |D|d <dD]} || rf|t| d|d<|SqR||d<|S)N)r#r$r%r'r&<iiQ)smhdtimeoutSecondsr(cSsg|]}|qSr)to_dict)r7volrrrr9sz$_create_job_spec..r))zhttps://huggingface.co/spaces/zhttps://hf.co/spaces/zhuggingface.co/spaces/z hf.co/spaces/r1r0)r CPU_BASIC isinstancerrHri startswithlen) rjr#rkr&r'rEr(r)rRtime_units_factorsprefixrrr_create_job_specls0 &  r|) dataclassesrrenumrtypingrhuggingface_hubrhuggingface_hub._space_apirrhuggingface_hub.utils._datetimer rr rrrrDrIrLrQrXr]rBrCrHrir|rrrrs^     a  C?