ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-12 15:45:25 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix uploading 3D plots with matplotlib plt shows 2D plot on task results page

Browse Source
This commit is contained in:
allegroai 2022-04-09 14:20:58 +03:00
parent 260690b990
commit 55f5f198ec
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
clearml/utilities/plotlympl/mplexporter/exporter.py
View File

@ -135,7 +135,7 @@ class Exporter(object):
self.draw_text(ax, text) self.draw_text(ax, text)
for (text, ttp) in zip([ax.xaxis.label, ax.yaxis.label, ax.title], for (text, ttp) in zip([ax.xaxis.label, ax.yaxis.label, ax.title],
["xlabel", "ylabel", "title"]): ["xlabel", "ylabel", "title"]):
if(hasattr(text, 'get_text') and text.get_text()): if(hasattr(text, "get_text") and text.get_text()):
self.draw_text(ax, text, force_trans=ax.transAxes, self.draw_text(ax, text, force_trans=ax.transAxes,
text_type=ttp) text_type=ttp)
for artist in ax.artists: for artist in ax.artists:
@ -246,6 +246,8 @@ class Exporter(object):
"""Process a matplotlib collection and call renderer.draw_collection""" """Process a matplotlib collection and call renderer.draw_collection"""
(transform, transOffset, (transform, transOffset,
offsets, paths) = collection._prepare_points() offsets, paths) = collection._prepare_points()
if hasattr(collection, "_offsets3d"):
offsets = collection._offsets3d
offset_coords, offsets = self.process_transform( offset_coords, offsets = self.process_transform(
transOffset, ax, offsets, force_trans=force_offsettrans) transOffset, ax, offsets, force_trans=force_offsettrans)