From 900ee9de5acea33d178ee6aed82b6d761dc3d3e6 Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Fri, 23 Jul 2021 15:57:38 +0300
Subject: [PATCH] Fix Pandas with multi-index (issue #399)

---
 clearml/utilities/plotly_reporter.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/clearml/utilities/plotly_reporter.py b/clearml/utilities/plotly_reporter.py
index d9fc5f06..515b9fd8 100644
--- a/clearml/utilities/plotly_reporter.py
+++ b/clearml/utilities/plotly_reporter.py
@@ -503,8 +503,12 @@ def create_plotly_table(table_plot, title, series, layout_config=None):
         headers_values = list([col] for col in table_plot.columns)
         cells_values = table_plot.T.values.tolist()
         if index_added:
-            headers_values.insert(0, table_plot.index.name or "")
-            cells_values.insert(0, table_plot.index.values.tolist())
+            if isinstance(table_plot.index, pd.MultiIndex):
+                headers_values = list(table_plot.index.names or []) + headers_values
+                cells_values = list(zip(*(table_plot.index.values.tolist() or []))) + cells_values
+            else:
+                headers_values.insert(0, table_plot.index.name or "")
+                cells_values.insert(0, table_plot.index.values.tolist())
 
     ret = {
         "data": [{