ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-03-03 10:42:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix multiple matplotlib windows with Agg backend interactive mode, caused duplicate graphs to be sent.

Browse Source
This commit is contained in:
allegroai 2019-09-27 13:22:17 +03:00
parent b720895307
commit 3eba23f1a3
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
trains/binding/matplotlib_bind.py
View File

@ -134,7 +134,8 @@ class PatchedMatplotlib:
# store on the plot that this is an imshow plot
stored_figure = _pylab_helpers.Gcf.get_active()
if stored_figure:
stored_figure._trains_is_imshow = True
stored_figure._trains_is_imshow = 1 if not hasattr(stored_figure, '_trains_is_imshow') \
else stored_figure._trains_is_imshow + 1
except Exception:
pass
return ret
@ -160,7 +161,9 @@ class PatchedMatplotlib:
try:
figures = PatchedMatplotlib._get_output_figures(None, all_figures=True)
for figure in figures:
PatchedMatplotlib._report_figure(stored_figure=figure)
# if this is a stale figure (just updated) we should send it, the rest will not be stale
if figure.canvas.figure.stale or (hasattr(figure, '_trains_is_imshow') and figure._trains_is_imshow):
PatchedMatplotlib._report_figure(stored_figure=figure)
except Exception:
pass
ret = PatchedMatplotlib._patched_original_plot(*args, **kw)
@ -193,9 +196,10 @@ class PatchedMatplotlib:
# nothing for us to do
return
# check if this is an imshow
if hasattr(stored_figure, '_trains_is_imshow') and stored_figure._trains_is_imshow:
force_save_as_image = True
if hasattr(stored_figure, '_trains_is_imshow'):
# flag will be cleared when calling clf() (object will be replaced)
stored_figure._trains_is_imshow = max(0, stored_figure._trains_is_imshow-1)
force_save_as_image = True
# get current figure
mpl_fig = stored_figure.canvas.figure # plt.gcf()
else: