From 6b8f21b30e005d50da35e2e5a5ded8782bce2f53 Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Sun, 24 Jan 2021 09:13:32 +0200
Subject: [PATCH] Add support for datetime in scatter plots + matplotlib
 support

---
 clearml/binding/matplotlib_bind.py | 10 ++++++++--
 clearml/logger.py                  |  2 +-
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/clearml/binding/matplotlib_bind.py b/clearml/binding/matplotlib_bind.py
index 80013de5..f670e971 100644
--- a/clearml/binding/matplotlib_bind.py
+++ b/clearml/binding/matplotlib_bind.py
@@ -372,10 +372,13 @@ class PatchedMatplotlib:
                             except Exception:
                                 # noinspection PyBroadException
                                 try:
-                                    plotly_renderer.plotly_fig['layout']['xaxis1'].update({
+                                    _xaxis = next(x for x in ('xaxis', 'xaxis0', 'xaxis1')
+                                                  if x in plotly_renderer.plotly_fig['layout'])
+                                    plotly_renderer.plotly_fig['layout'][_xaxis].update({
                                         'ticktext': [t.get_text() for t in x_ticks],
                                         'tickvals': [t.get_position()[0] for t in x_ticks],
                                     })
+                                    plotly_renderer.plotly_fig['layout'][_xaxis].pop('type', None)
                                 except Exception:
                                     pass
                         y_ticks = list(plotly_renderer.current_mpl_ax.get_yticklabels())
@@ -387,10 +390,13 @@ class PatchedMatplotlib:
                             except Exception:
                                 # noinspection PyBroadException
                                 try:
-                                    plotly_renderer.plotly_fig['layout']['yaxis1'].update({
+                                    _yaxis = next(x for x in ('yaxis', 'yaxis0', 'yaxis1')
+                                                  if x in plotly_renderer.plotly_fig['layout'])
+                                    plotly_renderer.plotly_fig['layout']['_yaxis'].update({
                                         'ticktext': [t.get_text() for t in y_ticks],
                                         'tickvals': [t.get_position()[1] for t in y_ticks],
                                     })
+                                    plotly_renderer.plotly_fig['layout'][_yaxis].pop('type', None)
                                 except Exception:
                                     pass
                         return deepcopy(plotly_renderer.plotly_fig)
diff --git a/clearml/logger.py b/clearml/logger.py
index b89880dd..3dd1fd67 100644
--- a/clearml/logger.py
+++ b/clearml/logger.py
@@ -485,7 +485,7 @@ class Logger(object):
         return self._task._reporter.report_2d_scatter(
             title=title,
             series=series,
-            data=scatter.astype(np.float32),
+            data=scatter,
             iter=iteration,
             mode=mode,
             xtitle=xaxis,