o -wi @sbddlZddlmZmZddlZGdddeZGdddeZGdddeZGd d d ZdS) N)MutableMapping NamedTuplec@s6eZdZUeed<eed<eed<eed<eed<dS) FileStatsdedupedtotaluploadedfailed artifact_fileN)__name__ __module__ __qualname__bool__annotations__intrrQ/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/wandb/filesync/stats.pyrs  rc@s&eZdZUeed<eed<eed<dS)Summaryuploaded_bytes total_bytes deduped_bytesNr r r rrrrrrrs  rc@s.eZdZUeed<eed<eed<eed<dS)FileCountsByCategoryartifactwandbmediaotherNrrrrrrs  rc @seZdZdddZ ddedededdfd d Zdeddfd d Zded eddfddZ deddfddZ de fddZ de fddZdS)StatsreturnNcCsi|_t|_dSN)_stats threadingLock_lock)selfrrr__init__szStats.__init__F save_namesizeis_artifact_filecCsD|jtd|dd|d|j|<WddS1swYdS)NFr)rrrrr )r"rr)r#r%r&r'rrr init_file!s"zStats.init_filecCsL|j|j|}|jd|jd|j|<WddS1swYdS)NT)rr)r"r_replacer)r#r%origrrrset_file_deduped-s "zStats.set_file_dedupedtotal_uploadedcCsD|j|j|j|d|j|<WddS1swYdS)Nrr"rr))r#r%r,rrrupdate_uploaded_file5s  "zStats.update_uploaded_filecCsF|j|j|jddd|j|<WddS1swYdS)NrT)rrr.)r#r%rrrupdate_failed_file;s  "zStats.update_failed_filecCsl|jt|j}Wdn1swYttdd|Dtdd|Dtdd|DdS)Ncs|]}|jVqdSrr-.0frrr Hz Stats.summary..csr1r)rr2rrrr5Ir6css|] }|jr|jVqdSr)rrr2rrrr5Js)rrr)r"listrvaluesrsum)r#statsrrrsummaryBsz Stats.summarycCsd}d}d}d}|jt|j}Wdn1swY|D]&\}}|jr0|d7}q$tjj|r<|d7}q$| drF|d7}q$|d7}q$t ||||dS)Nrr)rrrr) r"r7ritemsr r wandb_lib filenames is_wandb_file startswithr)r#artifact_files wandb_files media_files other_files file_statsr%r:rrrfile_counts_by_categoryMs*      zStats.file_counts_by_category)rN)F)r r r r$strrr r(r+r/r0rr;rrGrrrrrs"    r) r typingrrrrrrrrrrrs