1Vh7."dZddlZddlmZddlmZddlmZddlmZddl m Z ddl m Z e jZ ejd ZeZd Zd ZGd d ZGddZGddeZGddeZGddeZy)z4Functionality for loading events from a record file.N) data_compat)dataclass_compat)tf) event_pb2) platform_util) tb_loggingc#Kdyw)z=Pre-Python-3.7-compatible standin for contextlib.nullcontext.Nr f/home/dcms/DCMS/lib/python3.12/site-packages/tensorboard/backend/event_processing/event_file_loader.py _nullcontextr s  scf ddlm}|jS#ttf$r t cYSwxYw)zBContext manager that best-effort silences TF deprecation warnings.r deprecation)tensorflow.python.utilrsilence ImportErrorAttributeError _NULLCONTEXTrs r _silence_deprecation_warningsr)s36""$$  (s 00c tjdk(r:tjd|t tj j |S ddlm}|j }|r"tjd|t ||Stjd|t5tjjjj|cdddS#ttf$rd}YwxYw#1swYyxYw)z@Returns an iterator over TF records for the given tfrecord file.stubz+Opening a stub record reader pointing at %sr)pywrap_tensorflowNz'Opening a PyRecordReader pointing at %sz+Opening a tf_record_iterator pointing at %s)r __version__loggerdebug_PyRecordReaderIteratorrPyRecordReader_Newtensorflow.pythonrrrcompatv1iotf_record_iterator) file_pathrpy_record_reader_news r _make_tf_record_iteratorr&4s ~~  BIN&  3 3Y  $70CC > J&';YGG BIN * , A99<>**!  0$//B99**  :DOOQO    +T__=s<CE4CE4)EE4 E%!E E4 E$E4c:|j} tjjj |j j |_tjd|j |j||y|j|kDry|j|kr3tjd|j ||j||_y#tjj$r+}tjd|j |Yd}~yd}~wwxYw)aStats the file to get its updated size, returning True if it grew. If the stat call fails or reports a smaller size than was previously seen, then any previously cached size is left unchanged. Returns: boolean or None: True if the file size increased; False if it was the same or decreased; or None if neither case could be detected (either because the previous size had not been recorded yet, or because the stat call for the current size failed). zFailed to stat %s: %sNz(Stat on %s got size %d, previous size %sTz/File %s shrank from previous size %d to size %dF) rIrr"gfilestatrGlengthr*OpErrorrerrorrrL)r/ previous_sizer:s r rOz,RawEventFileLoader.CheckForIncreasedFileSizes   eekk..t?FFDO  6 OO OO     ??] * ??] * NNA  ,DO1yy    LL0$//1 E sACD/!DDN)F)r<r=r>r?r1rSrOr r r rBrBusI26)>V'r rBc"eZdZdZfdZxZS)LegacyEventFileLoaderz,An iterator that yields parsed Event protos.c#vKt|D]#}tjj |%yw)zLoads all new events from disk. Calling Load multiple times in a row will not 'drop' events as long as the return value is not iterated over. Yields: All events in the file that have not been yielded yet. N)superrSrEvent FromString)r/r9 __class__s r rSzLegacyEventFileLoader.Loads3gln 5F//,,V4 4 5s69r<r=r>r?rS __classcell__ras@r r\r\s6 5 5r r\c,eZdZdZfdZfdZxZS)EventFileLoaderzAn iterator that passes events through read-time compat layers. Specifically, this includes `data_compat` and `dataclass_compat`. c2t||i|i|_yr3)r^r1_initial_metadata)r/argskwargsras r r1zEventFileLoader.__init__s $)&)"$r c#Kt|D]B}tj|}t j||j }|D]}|Dywr3)r^rSr migrate_eventrrh)r/eventeventsras r rSzEventFileLoader.Load s]W\^ E--e4E%33t--F      sAA)r<r=r>r?r1rSrcrds@r rfrfs $r rfc"eZdZdZfdZxZS)TimestampedEventFileLoaderzBAn iterator that yields (UNIX timestamp float, Event proto) pairs.c#TKt|D]}|j|fyw)aOLoads all new events and their wall time values from disk. Calling Load multiple times in a row will not 'drop' events as long as the return value is not iterated over. Yields: Pairs of (UNIX timestamp float, Event proto) for all events in the file that have not been yielded yet. N)r^rS wall_time)r/rmras r rSzTimestampedEventFileLoader.Loads-W\^ +E??E* * +s%(rbrds@r rprpsL + +r rp)r? contextlib tensorboardrrtensorboard.compatrtensorboard.compat.protortensorboard.utilrr get_loggerrcontextmanagerr rrr&rrBr\rfrpr r r rzs ;#(!.*'       ~ AB@ppf 5. 5 +>++r