o
    w¡·iü
  ã                   @  sr   d dl mZ d dlZd dlmZ d dl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 )
é    )ÚannotationsN)ÚIterable)ÚAnyÚLiteralÚProtocolÚruntime_checkable)ÚArtifactÚimport_logger)ÚnowÚendÚlivec                   @  s.  e Zd ZdGdd„ZdGdd„ZdGdd„ZdHd
d„ZdIdd„ZdJdd„ZdKdd„Z	dKdd„Z
dGdd„ZdKdd„ZdLdd„ZdMdd „ZdMd!d"„ZdKd#d$„ZdKd%d&„ZdKd'd(„ZdKd)d*„ZdKd+d,„ZdKd-d.„ZdKd/d0„ZdKd1d2„ZdNd4d5„ZdNd6d7„ZdNd8d9„ZdNd:d;„ZdKd<d=„ZdKd>d?„ZdOdAdB„ZdPdDdE„ZdFS )QÚImporterRunÚreturnÚstrc                 C  ó   d S ©N© ©Úselfr   r   ú\/home/ubuntu/.local/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.projectúdict[str, Any]c                 C  r   r   r   r   r   r   r   Úconfig   r   zImporterRun.configúdict[str, float]c                 C  r   r   r   r   r   r   r   Úsummary   r   zImporterRun.summaryúIterable[dict[str, float]]c                 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.metricsú
str | Nonec                 C  r   r   r   r   r   r   r   Ú	run_group2   r   zImporterRun.run_groupc                 C  r   r   r   r   r   r   r   Újob_type4   r   zImporterRun.job_typec                 C  r   r   r   r   r   r   r   Údisplay_name6   r   zImporterRun.display_namec                 C  r   r   r   r   r   r   r   Únotes8   r   zImporterRun.notesúlist[str] | Nonec                 C  r   r   r   r   r   r   r   Útags:   r   zImporterRun.tagsúIterable[Artifact] | Nonec                 C  r   r   r   r   r   r   r   Ú	artifacts<   r   zImporterRun.artifactsc                 C  r   r   r   r   r   r   r   Úused_artifacts>   r   zImporterRun.used_artifactsc                 C  r   r   r   r   r   r   r   Ú
os_version@   r   zImporterRun.os_versionc                 C  r   r   r   r   r   r   r   Úpython_versionB   r   zImporterRun.python_versionc                 C  r   r   r   r   r   r   r   Úcuda_versionD   r   zImporterRun.cuda_versionc                 C  r   r   r   r   r   r   r   ÚprogramF   r   zImporterRun.programc                 C  r   r   r   r   r   r   r   ÚhostH   r   zImporterRun.hostc                 C  r   r   r   r   r   r   r   ÚusernameJ   r   zImporterRun.usernamec                 C  r   r   r   r   r   r   r   Ú
executableL   r   zImporterRun.executablec                 C  r   r   r   r   r   r   r   Ú	gpus_usedN   r   zImporterRun.gpus_usedú
int | Nonec                 C  r   r   r   r   r   r   r   Ú	cpus_usedP   r   zImporterRun.cpus_usedc                 C  r   r   r   r   r   r   r   Úmemory_usedR   r   zImporterRun.memory_usedc                 C  r   r   r   r   r   r   r   ÚruntimeT   r   zImporterRun.runtimec                 C  r   r   r   r   r   r   r   Ú
start_timeV   r   zImporterRun.start_timec                 C  r   r   r   r   r   r   r   Ú	code_pathX   r   zImporterRun.code_pathc                 C  r   r   r   r   r   r   r   Úcli_versionZ   r   zImporterRun.cli_versionú'Iterable[tuple[PathStr, Policy]] | Nonec                 C  r   r   r   r   r   r   r   Úfiles\   s   zImporterRun.filesúIterable[str] | Nonec                 C  r   r   r   r   r   r   r   Úlogs`   r   zImporterRun.logsN)r   r   )r   r   )r   r   )r   r   )r   r    )r   r%   )r   r'   )r   r2   )r   r9   )r   r;   ) Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r   r!   r"   r#   r$   r&   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   r3   r4   r5   r6   r7   r8   r:   r<   r   r   r   r   r      s<    



























r   )Ú
__future__r   ÚloggingÚcollections.abcr   Útypingr   r   r   r   Úwandb.sdk.artifacts.artifactr   Ú	getLoggerÚloggerr   ÚPathStrÚPolicyr   r   r   r   r   Ú<module>   s    
