o -wi @sUddlZddlZddlmZddlmZmZmZmZm Z m Z m Z ddl Z ddl mZddl mZer9ddlmZdZeed <Gd d d Z dd e d de edeefddZedZdeedeeeffddZddede efddZgdZdS)N) TracebackType) TYPE_CHECKINGContextManagerDictListOptionalSetType)Imports)TelemetryRecord) wandb_runz @wandbcode{ _LABEL_TOKENc@s~eZdZUeded<eed<  ddeddeeddfdd Zdefd d Zd eee d ee dee ddfddZ dS)_TelemetryObject wandb_run.Run_run_objNrunobjreturncCs|ptj|_|p t|_dSN)wandbrrr r)selfrrrT/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/wandb/sdk/lib/telemetry.py__init__s z_TelemetryObject.__init__cCs|jSr)r)rrrr __enter__sz_TelemetryObject.__enter__exctypeexcinstexctbcCs|jsdS|j|jdSr)r_telemetry_callbackr)rrrrrrr__exit__"sz_TelemetryObject.__exit__NN) __name__ __module__ __qualname__r__annotations__r rrr BaseExceptionrr!rrrrrs,   rrrrrcCs t||dS)Nrr)rr(rrrcontext-s r)z((?P[a-zA-Z0-9_-]+)[,}](?P.*)linesc Csd}i}|D]L}|t}|dkr|r|Sqd}||ttd}t|}|r:|ddd|d<|d}td|}|D]\}} | d dd||<qBq|S) NFrTcode-_restz;([a-zA-Z0-9_]+)\s*=\s*("[a-zA-Z0-9_-]*"|[a-zA-Z0-9_-]*)[,}]") findrlenMATCH_REmatchgroupreplacerefindallstrip) r*seenretlineidx label_strrtokenskvrrr_parse_label_lines6s,    rBF only_importedcCs.ddtjjD}|rttj}||S|S)NcSsh|] }|j|jkr|jqSr)type TYPE_BOOLname).0descrrr Ts  z)list_telemetry_imports..)TelemetryImports DESCRIPTORfieldssetsysmodules intersection)rCimport_telemetry_setimported_modules_setrrrlist_telemetry_importsSs  rS)rJr r)rSr")F)r6rNtypesrtypingrrrrrrr rwandb.proto.wandb_telemetry_pb2r rJr r rstrr&rr)compiler2rBboolrS__all__rrrrs.  $