a d=icJã@sTdZddlZddlZddlmZddlmZddlmZe ¡Z Gdd„de ƒZ dS)zCLoader for event file data for an entire TensorBoard log directory.éN)Údirectory_watcher)Ú io_wrapper)Ú tb_loggingc@s0eZdZdZdd„Zdd„Zdd„Zdd „Zd S) Ú LogdirLoaderaLoader for a root log directory, maintaining multiple DirectoryLoaders. This class takes a root log directory and a factory for DirectoryLoaders, and maintains one DirectoryLoader per "logdir subdirectory" of the root logdir. Note that this class is not thread-safe. cCs6|durtdƒ‚|dur tdƒ‚||_||_i|_dS)aZConstructs a new LogdirLoader. Args: logdir: The root log directory to load from. directory_loader_factory: A factory for creating DirectoryLoaders. The factory should take a path and return a DirectoryLoader. Raises: ValueError: If logdir or directory_loader_factory are None. NzA logdir is requiredz&A directory loader factory is required)Ú ValueErrorÚ_logdirÚ_directory_loader_factoryÚ_directory_loaders)ÚselfZlogdirZdirectory_loader_factory©r ús/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/tensorboard/uploader/logdir_loader.pyÚ__init__&s zLogdirLoader.__init__cCs¨t d|j¡tƒ}t |j¡D]D}tj ||j¡}|  |¡||j vr t d|¡|  |¡|j |<q t|j ƒ|}|r–|D]}t d|¡|j |=q|t d|j¡dS)zÉFinds new runs within `logdir` and makes `DirectoryLoaders` for them. In addition, any existing `DirectoryLoader` whose run directory no longer exists will be deleted. zStarting logdir traversal of %sz&- Adding run for relative directory %sz(- Removing run for relative directory %szEnding logdir traversal of %sN) ÚloggerÚinforÚsetrZGetLogdirSubdirectoriesÚosÚpathÚrelpathÚaddr r)r Z runs_seenÚsubdirÚrunZ stale_runsr r r Úsynchronize_runs:s    ÿ   zLogdirLoader.synchronize_runscCsPt|jƒ}t dt|ƒ¡t ¡}t|ƒD] }|j|}| |  ¡¡||<q*|S)aéReturns tf.Event generators for each run's `DirectoryLoader`. Warning: the generators are stateful and consuming them will affect the results of any other existing generators for that run; calling code should ensure it takes events from only a single generator per run at a time. Returns: Dictionary containing an entry for each run, mapping the run name to a generator yielding tf.Event protobuf objects loaded from that run. z-Creating event loading generators for %d runs) Úlistr rrÚlenÚ collectionsÚ OrderedDictÚsortedÚ_wrap_loader_generatorÚLoad)r ÚrunsZ run_to_loaderÚrun_nameÚloaderr r r Úget_run_eventsRs   zLogdirLoader.get_run_eventsccs0z|D] }|VqWntjy*YdS0dS)zMWraps `DirectoryLoader` generator to swallow `DirectoryDeletedError`.N)rZDirectoryDeletedError)r Zloader_generatorÚitemr r r res  z#LogdirLoader._wrap_loader_generatorN)Ú__name__Ú __module__Ú __qualname__Ú__doc__r rr"rr r r r rs r) r'rrZ$tensorboard.backend.event_processingrrZtensorboard.utilrZ get_loggerrÚobjectrr r r r Ús