o
    ni%                     @   s0   d dl mZ d dlmZ G dd dZdgZdS )    )Any)Apic                   @   sr  e Zd ZdZdededdfddZdd	 Zd
d Zede	fddZ
edd Zedd Zedd Zedd Zedd Zedd Zdd Zdd Zdd Zd d! Zd"d# Zd$d% Zd&d' Zdd(d)Zd*d+ Zdd,d-Zd.d/ Zd0d1 Z	2dd3ed4ed5eddfd6d7Z	2dd3ed8ed4ed5eddf
d9d:Z d;d< Z!d=d> Z"d?d@ Z#defdAdBZ$dCdD Z%dEdF Z&dGdH Z'dIdJ Z(dKdL Z)dMdN Z*dOdP Z+dQdR Z,dSdT Z-dUdV Z.dWdX Z/dYdZ Z0d[d\ Z1d]d^ Z2d_d` Z3dadb Z4dcdd Z5dedf Z6dgdh Z7didj Z8dkdl Z9dmdn Z:dodp Z;dqdr Z<dsdt Z=dudv Z>dwdx Z?dydz Z@d{d| ZAd}d~ ZBdd ZCdd ZDdd ZEdd ZFdd ZGdS )r   z,Internal proxy to the official internal API.argskwargsreturnNc                 O   s   || _ || _d | _d S N)	_api_args_api_kwargs_apiselfr   r    r   Q/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/wandb/apis/internal.py__init__   s   
zApi.__init__c                 C   s   | j  }|d= |S )zUUse for serializing.

        self._api is not serializable, so it's dropped
        r
   )__dict__copyr   stater   r   r   __getstate__   s   
zApi.__getstate__c                 C   s   | j | d| _dS )zkUsed for deserializing.

        Don't need to set self._api because it's constructed when needed.
        N)r   updater
   r   r   r   r   __setstate__   s   
zApi.__setstate__c                 C   s$   | j d u rt| ji | j| _ | j S r   )r
   InternalApir   r	   r   r   r   r   api!   s   
zApi.apic                 C      | j jS r   )r   api_keyr   r   r   r   r   *      zApi.api_keyc                 C   s   | j jd up| j jd uS r   )r   access_tokenr   r   r   r   r   is_authenticated.   s   zApi.is_authenticatedc                 C   r   r   )r   api_urlr   r   r   r   r   2   r   zApi.api_urlc                 C   r   r   )r   app_urlr   r   r   r   r    6   r   zApi.app_urlc                 C   r   r   )r   default_entityr   r   r   r   r!   :   r   zApi.default_entityc                 C   r   r   )r   gitr   r   r   r   r"   >   r   zApi.gitc                 G   s   | j j| S r   )r   file_current)r   r   r   r   r   r#   B      zApi.file_currentc                 O      | j j|i |S r   )r   download_filer   r   r   r   r&   E      zApi.download_filec                 O   r%   r   )r   download_write_filer   r   r   r   r(   H   r'   zApi.download_write_filec                 C      | j |S r   )r   set_current_run_id)r   run_idr   r   r   r*   K   r$   zApi.set_current_run_idc                 C   
   | j  S r   )r   viewerr   r   r   r   r-   N      
z
Api.viewerc                 C   r,   r   )r   max_cli_versionr   r   r   r   r/   Q   r.   zApi.max_cli_versionc                 C   r,   r   )r   viewer_server_infor   r   r   r   r0   T   r.   zApi.viewer_server_infoc                 C   s   | j j|dS )N)entity)r   list_projects)r   r1   r   r   r   r2   W   s   zApi.list_projectsc                 C   r)   r   )r   format_project)r   projectr   r   r   r3   Z   r$   zApi.format_projectc                 C   s   | j j||||dS )N)iddescriptionr1   )r   upsert_project)r   r4   r5   r6   r1   r   r   r   r7   ]   s   zApi.upsert_projectc                 O   r%   r   )r   
upsert_runr   r   r   r   r8   b   r'   zApi.upsert_runc                 O   r%   r   )r   settingsr   r   r   r   r9   e   r'   zApi.settingsFkeygloballypersistc                 C   s   | j |||S r   )r   clear_setting)r   r:   r;   r<   r   r   r   r=   h   s   zApi.clear_settingvaluec                 C   s   | j ||||S r   )r   set_setting)r   r:   r>   r;   r<   r   r   r   r?   m   s   zApi.set_settingc                 O   r%   r   )r   
parse_slugr   r   r   r   r@   r   r'   zApi.parse_slugc                 O   r%   r   )r   download_urlr   r   r   r   rA   u   r'   zApi.download_urlc                 O   r%   r   )r   download_urlsr   r   r   r   rB   x   r'   zApi.download_urlsc                 C   r,   r   )r   create_anonymous_api_keyr   r   r   r   rC   {   r.   zApi.create_anonymous_api_keyc                 O   r%   r   )r   pushr   r   r   r   rD   ~   r'   zApi.pushc                 O   r%   r   )r   sweepr   r   r   r   rE      r'   z	Api.sweepc                 O   r%   r   )r   upsert_sweepr   r   r   r   rF      r'   zApi.upsert_sweepc                 O   r%   r   )r   set_sweep_stater   r   r   r   rG      r'   zApi.set_sweep_statec                 O   r%   r   )r   get_sweep_stater   r   r   r   rH      r'   zApi.get_sweep_statec                 O   r%   r   )r   
stop_sweepr   r   r   r   rI      r'   zApi.stop_sweepc                 O   r%   r   )r   cancel_sweepr   r   r   r   rJ      r'   zApi.cancel_sweepc                 O   r%   r   )r   pause_sweepr   r   r   r   rK      r'   zApi.pause_sweepc                 O   r%   r   )r   resume_sweepr   r   r   r   rL      r'   zApi.resume_sweepc                 O   r%   r   )r   register_agentr   r   r   r   rM      r'   zApi.register_agentc                 O   r%   r   )r   agent_heartbeatr   r   r   r   rN      r'   zApi.agent_heartbeatc                 O   r%   r   )r   use_artifactr   r   r   r   rO      r'   zApi.use_artifactc                 O   r%   r   )r   create_artifactr   r   r   r   rP      r'   zApi.create_artifactc                 O   r%   r   )r   "complete_multipart_upload_artifactr   r   r   r   rQ      r'   z&Api.complete_multipart_upload_artifactc                 O   r%   r   )r   
run_configr   r   r   r   rR      r'   zApi.run_configc                 O   r%   r   )r   upload_file_retryr   r   r   r   rS      r'   zApi.upload_file_retryc                 O   r%   r   )r   !upload_multipart_file_chunk_retryr   r   r   r   rT      r'   z%Api.upload_multipart_file_chunk_retryc                 O   r%   r   )r   get_run_infor   r   r   r   rU      r'   zApi.get_run_infoc                 O   r%   r   )r   get_run_stater   r   r   r   rV      r'   zApi.get_run_statec                 O   r%   r   )r   entity_is_teamr   r   r   r   rW      r'   zApi.entity_is_teamc                 O   r%   r   )r   get_project_run_queuesr   r   r   r   rX      r'   zApi.get_project_run_queuesc                 O   r%   r   )r   push_to_run_queuer   r   r   r   rY      r'   zApi.push_to_run_queuec                 O   r%   r   )r   pop_from_run_queuer   r   r   r   rZ      r'   zApi.pop_from_run_queuec                 O   r%   r   )r   ack_run_queue_itemr   r   r   r   r[      r'   zApi.ack_run_queue_itemc                 O   r%   r   )r   create_launch_agentr   r   r   r   r\      r'   zApi.create_launch_agentc                 O   r%   r   )r   create_default_resource_configr   r   r   r   r]      r'   z"Api.create_default_resource_configc                 O   r%   r   )r   create_run_queuer   r   r   r   r^      r'   zApi.create_run_queuec                 O   r%   r   )r   upsert_run_queuer   r   r   r   r_      r'   zApi.upsert_run_queuec                 O   r%   r   )r   update_launch_agent_statusr   r   r   r   r`      r'   zApi.update_launch_agent_statusc                 O   r%   r   )r   launch_agent_introspectionr   r   r   r   ra      r'   zApi.launch_agent_introspectionc                 O   r%   r   )r   !fail_run_queue_item_introspectionr   r   r   r   rb      r'   z%Api.fail_run_queue_item_introspectionc                 O   r%   r   )r   fail_run_queue_itemr   r   r   r   rc      r'   zApi.fail_run_queue_itemc                 O   r%   r   )r   update_run_queue_item_warningr   r   r   r   rd      r'   z!Api.update_run_queue_item_warningc                 O   r%   r   )r   get_launch_agentr   r   r   r   re      r'   zApi.get_launch_agentc                 O   r%   r   )r   stop_runr   r   r   r   rf      r'   zApi.stop_runr   )NNN)FF)H__name__
__module____qualname____doc__r   r   r   r   propertyr   r   r   r   r   r    r!   r"   r#   r&   r(   r*   r-   r/   r0   r2   r3   r7   r8   r9   strboolr=   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rQ   rR   rS   rT   rU   rV   rW   rX   rY   rZ   r[   r\   r]   r^   r_   r`   ra   rb   rc   rd   re   rf   r   r   r   r   r      s    	









r   N)typingr   wandb.sdk.internal.internal_apir   r   __all__r   r   r   r   <module>   s
     
c