ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-02 01:54:23 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Fix matplotlib bind load config only when needed (saves double loading at init)

Browse Source
This commit is contained in:
allegroai 2022-05-05 12:03:52 +03:00
parent 05094dd865
commit 6ce91e5288
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
clearml/binding/matplotlib_bind.py
View File

@ -189,12 +189,15 @@ class PatchedMatplotlib:
return True return True
@staticmethod @staticmethod
def update_current_task(task): def _get_global_image_counter_limit():
# make sure we have a default vale # make sure we have a default vale
if PatchedMatplotlib._global_image_counter_limit is None: if PatchedMatplotlib._global_image_counter_limit is None:
from ..config import config from ..config import config
PatchedMatplotlib._global_image_counter_limit = config.get('metric.matplotlib_untitled_history_size', 100) PatchedMatplotlib._global_image_counter_limit = config.get('metric.matplotlib_untitled_history_size', 100)
return PatchedMatplotlib._global_image_counter_limit
@staticmethod
def update_current_task(task):
# if we already patched it, just update the current task # if we already patched it, just update the current task
if PatchedMatplotlib._patched_original_plot is not None: if PatchedMatplotlib._patched_original_plot is not None:
PatchedMatplotlib._current_task = task PatchedMatplotlib._current_task = task
@ -535,11 +538,11 @@ class PatchedMatplotlib:
elif report_as_debug_sample: elif report_as_debug_sample:
PatchedMatplotlib._global_image_counter += 1 PatchedMatplotlib._global_image_counter += 1
title = 'untitled {:02d}'.format( title = 'untitled {:02d}'.format(
PatchedMatplotlib._global_image_counter % PatchedMatplotlib._global_image_counter_limit) PatchedMatplotlib._global_image_counter % PatchedMatplotlib._get_global_image_counter_limit())
else: else:
PatchedMatplotlib._global_plot_counter += 1 PatchedMatplotlib._global_plot_counter += 1
title = 'untitled {:02d}'.format( title = 'untitled {:02d}'.format(
PatchedMatplotlib._global_plot_counter % PatchedMatplotlib._global_image_counter_limit) PatchedMatplotlib._global_plot_counter % PatchedMatplotlib._get_global_image_counter_limit())
# by now we should have a title, if the iteration was known list us as globally reported. # by now we should have a title, if the iteration was known list us as globally reported.
# we later use it to check if externally someone was actually reporting iterations # we later use it to check if externally someone was actually reporting iterations