o xi @snddlZddlmZddlmZddlmZGdddeZGdddeZGd d d eZ Gd d d Z dS) N)MutableMapping) NamedTuple) filenamesc@s6eZdZUeed<eed<eed<eed<eed<dS) FileStatsdedupedtotaluploadedfailed artifact_fileN)__name__ __module__ __qualname__bool__annotations__intrrH/home/ubuntu/.local/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)rrrr r )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#r r*)r$r&r-rrrupdate_uploaded_file6s  "zStats.update_uploaded_filecCsF|j|j|jddd|j|<WddS1swYdS)NrT)rr r/)r$r&rrrupdate_failed_file<s  "zStats.update_failed_filecCsl|jt|j}Wdn1swYttdd|Dtdd|Dtdd|DdS)Ncs|]}|jVqdSrr..0frrr Iz Stats.summary..csr2r)rr3rrrr6Jr7css|] }|jr|jVqdSr)rrr3rrrr6Ks)rrr)r#listr valuesrsum)r$statsrrrsummaryCsz Stats.summarycCsd}d}d}d}|jt|j}Wdn1swY|D]$\}}|jr0|d7}q$t|r:|d7}q$|drD|d7}q$|d7}q$t||||dS)Nrr)rrrr) r#r8r itemsr r is_wandb_file startswithr)r$artifact_files wandb_files media_files other_files file_statsr&r;rrrfile_counts_by_categoryNs*       zStats.file_counts_by_category)rN)F)r r r r%strrrr)r,r0r1rr<rrFrrrrrs"    r) r!collections.abcrtypingr wandb.sdk.librrrrrrrrrs