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

Fix Matplotlib automagic export legend not showing series names (issue #337)

Browse Source
This commit is contained in:
allegroai 2021-04-20 18:08:50 +03:00
parent 4832de5a65
commit 22d795f68f
2 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
clearml/binding/matplotlib_bind.py
View File

@ -408,6 +408,22 @@ class PatchedMatplotlib:
plotly_renderer.plotly_fig['layout'][_yaxis].pop('type', None)
except Exception:
pass
# try to bring back legend
# noinspection PyBroadException
try:
if plotly_renderer.plotly_fig.get('layout', {}).get('showlegend') and \
plotly_renderer.plotly_fig.get('data'):
# the legend names is the upper half of the data points:
lines_ = plotly_renderer.plotly_fig['data']
half_mark = len(lines_)//2
if len(lines_) % 2 == 0 and \
all(l for l in lines_[half_mark:] if not l.get('x') and not l.get('y')):
for i, line in enumerate(lines_[:half_mark]):
line['name'] = lines_[i+half_mark].get('name')
except Exception:
pass
return deepcopy(plotly_renderer.plotly_fig)
plotly_dict = our_mpl_to_plotly(mpl_fig)

1
clearml/utilities/plotlympl/mpltools.py
View File

@ -156,6 +156,7 @@ def merge_color_and_opacity(color, opacity):
if opacity is None:
return "rgb {}".format(rgb_tup)
opacity = int(opacity*255.) if opacity <= 1.0 else int(opacity)
rgba_tup = rgb_tup + (opacity,)
return "rgba {}".format(rgba_tup)