o
    0Îj8  ã                   @   sŽ  d dl mZ d dlmZ d dlmZ d dlmZ d dlmZ d dl	m
Z
mZ d dlmZ G dd	„ d	eeƒZeG d
d„ dƒƒZeG dd„ dƒƒZeG dd„ dƒƒZeG dd„ dƒƒZeG dd„ dƒƒZeG dd„ dƒƒZeG dd„ dƒƒZeG dd„ dƒƒZeG dd„ dƒƒZdddœdedee d eeef dB d!eeef dB d"e
dB d#eeB eB dB d$eeef dB d%ee dB d&eeef fd'd(„ZdS ))é    )Ú	dataclass)Údatetime)ÚEnum)ÚAny)Ú	constants)ÚSpaceHardwareÚVolume)Úparse_datetimec                   @   s$   e Zd ZdZdZdZdZdZdZdS )Ú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__r   r   r   r   r   © r   r   úZ/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/huggingface_hub/_jobs_api.pyr
      s    r
   c                   @   s"   e Zd ZU eed< edB ed< dS )Ú	JobStatusÚstageNÚmessage)r   r   r   r
   Ú__annotations__Ústrr   r   r   r   r   ,   s   
 r   c                   @   s&   e Zd ZU eed< eed< eed< dS )ÚJobOwnerÚidÚnameÚtypeN)r   r   r   r   r   r   r   r   r   r   2   s   
 r   c                   @   sà   e Zd ZU dZeed< edB ed< edB ed< edB ed< ee dB ed< ee dB ed< eee	f dB ed	< eee	f dB ed
< e
dB ed< eeef dB ed< ee dB ed< eed< eed< eed< eed< ddd„ZdS )Ú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'
    ```
    r   NÚ
created_atÚdocker_imageÚspace_idÚcommandÚ	argumentsÚenvironmentÚsecretsÚflavorÚlabelsÚvolumesÚstatusÚownerÚendpointÚurlÚreturnc                 K   sD  |d | _ | d¡p| d¡}|rt|ƒnd | _| d¡p!| d¡| _| d¡p,| d¡| _| di ¡}t|d |d	 |d
 d| _| d¡| _| d¡| _	| d¡| _
| d¡| _| d¡| _| d¡| _| d¡}|rtdd„ |D ƒnd | _| di ¡}t|d | d¡d| _| dtj¡| _| j› d| jj› d| j › | _d S )Nr   Ú	createdAtr    ÚdockerImager!   ÚspaceIdr"   r+   r   r   ©r   r   r   r#   r$   r%   r&   r'   r(   r)   c                 S   ó   g | ]	}t d i |¤Ž‘qS ©r   ©r   ©Ú.0Úvr   r   r   Ú
<listcomp>’   ó    z$JobInfo.__init__.<locals>.<listcomp>r*   r   r   )r   r   r,   z/jobs/ú/)r   Úgetr	   r    r!   r"   r   r+   r#   r$   r%   r&   r'   r(   r)   r   r*   r   ZENDPOINTr,   r   r-   )ÚselfÚkwargsr    r+   r)   r*   r   r   r   Ú__init__ƒ   s&   

"zJobInfo.__init__©r.   N)r   r   r   r   r   r   r   ÚlistÚdictr   r   r   r   r   r?   r   r   r   r   r   9   s$   
 6r   c                   @   sÐ   e Zd ZU edB ed< edB ed< ee dB ed< ee dB ed< eeef dB ed< eeef dB ed< edB ed< e	dB ed	< ee dB ed
< edB ed< eeef dB ed< ee
 dB ed< ddd„ZdS )ÚJobSpecNr!   r"   r#   r$   r%   r&   r'   ÚtimeoutÚtagsÚarchr(   r)   r.   c                 K   sÄ   |  d¡p	|  d¡| _|  d¡p|  d¡| _|  d¡| _|  d¡| _|  d¡| _|  d¡| _|  d	¡| _|  d
¡| _|  d¡| _	|  d¡| _
|  d¡| _|  d¡}|r]dd„ |D ƒ| _d S d | _d S )Nr0   r!   r1   r"   r#   r$   r%   r&   r'   rD   rE   rF   r(   r)   c                 S   r3   r4   r5   r6   r   r   r   r9   ·   r:   z$JobSpec.__init__.<locals>.<listcomp>)r<   r!   r"   r#   r$   r%   r&   r'   rD   rE   rF   r(   r)   )r=   r>   r)   r   r   r   r?   ª   s   
"zJobSpec.__init__r@   )r   r   r   r   r   rA   rB   r   r   Úintr   r?   r   r   r   r   rC   ›   s   
 rC   c                   @   s(   e Zd ZU eed< eed< ddd„ZdS )ÚLastJobInfor   Úatr.   Nc                 K   s   |d | _ t|d ƒ| _d S )Nr   rI   )r   r	   rI   ©r=   r>   r   r   r   r?   ¿   s   
zLastJobInfo.__init__r@   )r   r   r   r   r   r   r?   r   r   r   r   rH   º   s   
 rH   c                   @   s0   e Zd ZU edB ed< edB ed< ddd„ZdS )ÚScheduledJobStatusNÚlast_jobÚnext_job_run_atr.   c                 K   s`   |  d¡p	|  d¡}|rtdi |¤Žnd | _|  d¡p|  d¡}|r+tt|ƒƒ| _d S d | _d S )NÚlastJobrL   ÚnextJobRunAtrM   r   )r<   rH   rL   r	   r   rM   )r=   r>   rL   rM   r   r   r   r?   É   s    zScheduledJobStatus.__init__r@   )r   r   r   rH   r   r   r?   r   r   r   r   rK   Ä   s   
 rK   c                   @   sl   e Zd ZU dZeed< edB ed< eed< edB ed< edB ed< edB ed< e	ed	< e
e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)
    ```
    r   Nr    Újob_specÚscheduleÚsuspendÚconcurrencyr*   r+   r.   c                 K   sÜ   |d | _ | d¡p| d¡}|rt|ƒnd | _tdi | d¡p%| di ¡¤Ž| _| d¡| _| d¡| _| d¡| _| d	i ¡}t	| d
¡pK| d¡| d¡pT| d¡d| _
| di ¡}t|d |d |d d| _d S )Nr   r/   r    rQ   ZjobSpecrR   rS   rT   r*   rL   rN   rM   rO   )rL   rM   r+   r   r   r2   r   )r   r<   r	   r    rC   rQ   rR   rS   rT   rK   r*   r   r+   )r=   r>   r    r*   r+   r   r   r   r?     s   
"þ zScheduledJobInfo.__init__r@   )r   r   r   r   r   r   r   rC   ÚboolrK   r   r?   r   r   r   r   rP   Ð   s   
 (rP   c                   @   sD   e Zd ZU dZeed< eed< eed< eed< eed< dd	d
„ZdS )Ú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"`.
    r   ÚmodelÚquantityÚvramÚmanufacturerr.   Nc                 K   s6   |d | _ |d | _|d | _|d | _|d | _d S )Nr   rW   rX   rY   rZ   )r   rW   rX   rY   rZ   rJ   r   r   r   r?   ,  s
   



zJobAccelerator.__init__r@   )r   r   r   r   r   r   r?   r   r   r   r   rV     s   
 rV   c                   @   s`   e Zd ZU dZeed< eed< eed< eed< edB ed< eed< eed	< eed
< ddd„Z	dS )Ú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_nameÚcpuÚramNÚacceleratorÚunit_cost_micro_usdÚunit_cost_usdÚ
unit_labelr.   c                 K   sl   |d | _ |d | _|d | _|d | _| d¡}|r"td	i |¤Žnd | _|d | _|d | _|d | _	d S )
Nr   Z
prettyNamer]   r^   r_   ZunitCostMicroUSDZunitCostUSDZ	unitLabelr   )
r   r\   r]   r^   r<   rV   r_   r`   ra   rb   )r=   r>   r_   r   r   r   r?   `  s   






zJobHardware.__init__r@   )
r   r   r   r   r   r   rV   rG   Úfloatr?   r   r   r   r   r[   4  s   
 !r[   N)r(   r)   Úimager#   Úenvr&   r'   rD   r(   r)   r.   c                 C   sÚ   |g |pi |p	t jdœ}|r||d< |r?dddddœ}	t|tƒr9|d |	v r9tt|d d… ƒ|	|d   ƒ|d	< nt|ƒ|d	< |rE||d
< |rPdd„ |D ƒ|d< dD ]}
|  |
¡rf| t|
ƒd … |d<  |S qR| |d< |S )N)r#   r$   r%   r'   r&   é   é<   i  i€Q )ÚsÚmÚhÚdéÿÿÿÿZtimeoutSecondsr(   c                 S   s   g | ]}|  ¡ ‘qS r   )Úto_dict)r7   Zvolr   r   r   r9     s    z$_create_job_spec.<locals>.<listcomp>r)   )zhttps://huggingface.co/spaces/zhttps://hf.co/spaces/zhuggingface.co/spaces/zhf.co/spaces/r1   r0   )r   Z	CPU_BASICÚ
isinstancer   rG   rc   Ú
startswithÚlen)rd   r#   re   r&   r'   rD   r(   r)   rQ   Ztime_units_factorsÚprefixr   r   r   Ú_create_job_specl  s0   ü&
ûrr   )Údataclassesr   r   Úenumr   Útypingr   Zhuggingface_hubr   Zhuggingface_hub._space_apir   r   Zhuggingface_hub.utils._datetimer	   r   r
   r   r   r   rC   rH   rK   rP   rV   r[   rA   rB   rG   rc   rr   r   r   r   r   Ú<module>   s^   a	C?÷þýüûúùø
	÷

ö