o
    -wÖiE  ã                   @   sn   d dl Z d dlmZmZmZmZmZmZmZm	Z	m
Z
 d dlmZ e  d¡ZeZed Ze
G dd„ deƒƒZdS )é    N)	ÚAnyÚDictÚIterableÚListÚLiteralÚOptionalÚProtocolÚTupleÚruntime_checkable)ÚArtifactÚimport_logger)ÚnowÚendÚlivec                   @   s2  e Zd Zdefdd„Zdefdd„Zdefdd„Zdeeef fdd	„Z	deee
f fd
d„Zdeeee
f  fdd„Zdee fdd„Zdee fdd„Zdefdd„Zdee fdd„Zdeee  fdd„Zdeee  fdd„Zdeee  fdd„Zdee fdd„Zdee fdd„Zdee fd d!„Zdee fd"d#„Zdee fd$d%„Zdee fd&d'„Zdee fd(d)„Zdee fd*d+„Zdee  fd,d-„Z!dee  fd.d/„Z"dee  fd0d1„Z#dee  fd2d3„Z$dee fd4d5„Z%dee fd6d7„Z&deee'e(e)f   fd8d9„Z*deee  fd:d;„Z+d<S )=ÚImporterRunÚreturnc                 C   ó   d S ©N© ©Úselfr   r   úe/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/wandb/apis/importers/internals/protocols.pyÚrun_id   ó    zImporterRun.run_idc                 C   r   r   r   r   r   r   r   Úentity   r   zImporterRun.entityc                 C   r   r   r   r   r   r   r   Úproject   r   zImporterRun.projectc                 C   r   r   r   r   r   r   r   Úconfig   r   zImporterRun.configc                 C   r   r   r   r   r   r   r   Úsummary    r   zImporterRun.summaryc                 C   s   dS )a  Metrics for the run.

        We expect metrics in this shape:

        [
            {'metric1': 1, 'metric2': 1, '_step': 0},
            {'metric1': 2, 'metric2': 4, '_step': 1},
            {'metric1': 3, 'metric2': 9, '_step': 2},
            ...
        ]

        You can also submit metrics in this shape:
        [
            {'metric1': 1, '_step': 0},
            {'metric2': 1, '_step': 0},
            {'metric1': 2, '_step': 1},
            {'metric2': 4, '_step': 1},
            ...
        ]
        Nr   r   r   r   r   Úmetrics"   s   zImporterRun.metricsc                 C   r   r   r   r   r   r   r   Ú	run_group9   r   zImporterRun.run_groupc                 C   r   r   r   r   r   r   r   Újob_type;   r   zImporterRun.job_typec                 C   r   r   r   r   r   r   r   Údisplay_name=   r   zImporterRun.display_namec                 C   r   r   r   r   r   r   r   Únotes?   r   zImporterRun.notesc                 C   r   r   r   r   r   r   r   ÚtagsA   r   zImporterRun.tagsc                 C   r   r   r   r   r   r   r   Ú	artifactsC   r   zImporterRun.artifactsc                 C   r   r   r   r   r   r   r   Úused_artifactsE   r   zImporterRun.used_artifactsc                 C   r   r   r   r   r   r   r   Ú
os_versionG   r   zImporterRun.os_versionc                 C   r   r   r   r   r   r   r   Úpython_versionI   r   zImporterRun.python_versionc                 C   r   r   r   r   r   r   r   Úcuda_versionK   r   zImporterRun.cuda_versionc                 C   r   r   r   r   r   r   r   ÚprogramM   r   zImporterRun.programc                 C   r   r   r   r   r   r   r   ÚhostO   r   zImporterRun.hostc                 C   r   r   r   r   r   r   r   ÚusernameQ   r   zImporterRun.usernamec                 C   r   r   r   r   r   r   r   Ú
executableS   r   zImporterRun.executablec                 C   r   r   r   r   r   r   r   Ú	gpus_usedU   r   zImporterRun.gpus_usedc                 C   r   r   r   r   r   r   r   Ú	cpus_usedW   r   zImporterRun.cpus_usedc                 C   r   r   r   r   r   r   r   Úmemory_usedY   r   zImporterRun.memory_usedc                 C   r   r   r   r   r   r   r   Úruntime[   r   zImporterRun.runtimec                 C   r   r   r   r   r   r   r   Ú
start_time]   r   zImporterRun.start_timec                 C   r   r   r   r   r   r   r   Ú	code_path_   r   zImporterRun.code_pathc                 C   r   r   r   r   r   r   r   Úcli_versiona   r   zImporterRun.cli_versionc                 C   r   r   r   r   r   r   r   Úfilesc   s   zImporterRun.filesc                 C   r   r   r   r   r   r   r   Úlogsg   r   zImporterRun.logsN),Ú__name__Ú
__module__Ú__qualname__Ústrr   r   r   r   r   r   Úfloatr   r   r   r   r   r    r!   r"   r   r#   r   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   Úintr.   r/   r0   r1   r2   r3   r	   ÚPathStrÚPolicyr4   r5   r   r   r   r   r      s@    
þr   )ÚloggingÚtypingr   r   r   r   r   r   r   r	   r
   Úwandb.sdk.artifacts.artifactr   Ú	getLoggerÚloggerr9   r<   r=   r   r   r   r   r   Ú<module>   s    ,
