U Ód¨iƒ*ã @sHddlmZddlmZddlmZddlmZmZmZm Z m Z ddl m Z ddl mZddlmZGdd „d eeƒZeGd d „d ƒƒZeGd d „d ƒƒZeGdd„dƒƒZeGdd„dƒƒZeGdd„dƒƒZeGdd„dƒƒZeGdd„dƒƒZeeee eeefe eeefe ee e eeefeeefdœdd„ZdS)é)Ú dataclass)Údatetime)ÚEnum)ÚAnyÚDictÚListÚOptionalÚUnion)Ú constants)Ú SpaceHardware)Ú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). Ú COMPLETEDÚCANCELEDÚERRORÚDELETEDÚRUNNINGN) Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrrr©rrú=/tmp/pip-unpacked-wheel-ohruig0i/huggingface_hub/_jobs_api.pyr s  r c@s"eZdZUeed<eeed<dS)Ú JobStatusÚstageÚmessageN)rrrr Ú__annotations__rÚstrrrrrr-s rc@s&eZdZUeed<eed<eed<dS)ÚJobOwnerÚidÚnameÚtypeN)rrrrrrrrrr3s rc@sÀeZdZUdZeed<eeed<eeed<eeed<eeeed<eeeed<ee ee fed<ee ee fed <ee ed <e ed <e ed <eed <eed<ddœdd„ZdS)Ú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"`. 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', 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' ``` rÚ created_atÚ docker_imageÚspace_idÚcommandÚ argumentsÚ environmentÚsecretsÚflavorÚstatusÚownerÚendpointÚurlN©ÚreturncKs|d|_| d¡p| d¡}|r*t|ƒnd|_| d¡pB| d¡|_| d¡pX| d¡|_| di¡}t|d|d |d d |_| d ¡|_| d ¡|_ | d¡|_ | d¡|_ | d¡|_ | di¡}t |d| d¡d|_| dtj¡|_|j›d|jj›d|j›|_dS)NrÚ createdAtr#Ú dockerImager$ÚspaceIdr%r,r r!©rr r!r&r'r(r)r*r+rr)rrr-z/jobs/ú/)rÚgetr r#r$r%rr,r&r'r(r)r*rr+r ZENDPOINTr-r r.)ÚselfÚkwargsr#r,r+rrrÚ__init__~s         zJobInfo.__init__)rrrrrrrrrrrr rrr9rrrrr":s 2    r"c@s°eZdZUeeed<eeed<eeeed<eeeed<eeeefed<eeeefed<ee ed<ee ed<eeeed <eeed <d d œd d„Z d S)ÚJobSpecr$r%r&r'r(r)r*ÚtimeoutÚtagsÚarchNr/cKs| d¡p| d¡|_| d¡p(| d¡|_| d¡|_| d¡|_| d¡|_| d¡|_| d ¡|_| d ¡|_| d ¡|_ | d ¡|_ dS) Nr2r$r3r%r&r'r(r)r*r;r<r=) r6r$r%r&r'r(r)r*r;r<r=©r7r8rrrr9 s       zJobSpec.__init__) rrrrrrrrrr Úintr9rrrrr:“s      r:c@s,eZdZUeed<eed<ddœdd„ZdS)Ú LastJobInforÚatNr/cKs|d|_t|dƒ|_dS)NrrA)rr rAr>rrrr9²s zLastJobInfo.__init__)rrrrrrr9rrrrr@­s r@c@s4eZdZUeeed<eeed<ddœdd„ZdS)ÚScheduledJobStatusÚlast_jobÚnext_job_run_atNr/cKsV| d¡p| d¡}|r"tf|Žnd|_| d¡p:| d¡}|rLtt|ƒƒnd|_dS)NÚlastJobrCÚ nextJobRunAtrD)r6r@rCr rrD)r7r8rCrDrrrr9¼szScheduledJobStatus.__init__)rrrrr@rrr9rrrrrB·s   rBc@speZdZUdZeed<eeed<eed<eeed<ee ed<ee ed<e ed<e ed <d d œd d „Z d S)Ú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) ``` rr#Újob_specÚscheduleÚsuspendÚ concurrencyr+r,Nr/cKsØ|d|_| d¡p| d¡}|r*t|ƒnd|_tf| d¡pH| di¡Ž|_| d¡|_| d¡|_| d¡|_| d i¡}t | d ¡p’| d ¡| d ¡p¤| d ¡d|_ | di¡}t |d|d|dd|_ dS)Nrr1r#rHZjobSpecrIrJrKr+rCrErDrF)rCrDr,r r!r4) rr6r r#r:rHrIrJrKrBr+rr,)r7r8r#r+r,rrrr9ös     þ zScheduledJobInfo.__init__) rrrrrrrrr:ÚboolrBrr9rrrrrGÃs (    rG)Úimager&Úenvr)r*r;r0c Cs¶|g|p i|ptjdœ}|r$||d<|r~dddddœ}t|tƒrr|d|krrtt|dd…ƒ||dƒ|d <n t|ƒ|d <d D]&}| |¡r‚|t|ƒd…|d <q²q‚||d <|S) N)r&r'r(r*r)éé<ii€Q)ÚsÚmÚhÚdéÿÿÿÿZtimeoutSeconds)zhttps://huggingface.co/spaces/zhttps://hf.co/spaces/zhuggingface.co/spaces/z hf.co/spaces/r3r2)r Z CPU_BASICÚ isinstancerr?ÚfloatÚ startswithÚlen) rMr&rNr)r*r;rHZtime_units_factorsÚprefixrrrÚ_create_job_specs$ ü&  r[N)Z dataclassesrrÚenumrÚtypingrrrrr Zhuggingface_hubr Zhuggingface_hub._space_apir Zhuggingface_hub.utils._datetimer rr rrr"r:r@rBrGr?rWr[rrrrÚs:      X  E ø