From ba32cad486b4df5dd792485d57babb1440289d5a Mon Sep 17 00:00:00 2001
From: Florian Sattler <vuld3r@gmail.com>
Date: Fri, 20 Oct 2023 13:57:16 +0200
Subject: [PATCH] Extends and fixes minor plotting/table issues

---
 varats/varats/plots/scatter_plot_utils.py | 3 ++-
 varats/varats/tables/time_workloads.py    | 6 +++++-
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/varats/varats/plots/scatter_plot_utils.py b/varats/varats/plots/scatter_plot_utils.py
index 3510d3da4..005dba2ed 100644
--- a/varats/varats/plots/scatter_plot_utils.py
+++ b/varats/varats/plots/scatter_plot_utils.py
@@ -13,6 +13,7 @@ def multivariate_grid(
     y: str,
     hue: str,
     global_kde: bool = True,
+    legend: bool = True,
     **kwargs: tp.Any
 ) -> sns.JointGrid:
     """
@@ -84,7 +85,7 @@ def multivariate_grid(
             color='grey',
             warn_singular=False
         )
-    if len(grouped_data) > 1:
+    if len(grouped_data) > 1 and legend:
         plt.legend(legends)
 
     return grid
diff --git a/varats/varats/tables/time_workloads.py b/varats/varats/tables/time_workloads.py
index 4120c5210..234e8254a 100644
--- a/varats/varats/tables/time_workloads.py
+++ b/varats/varats/tables/time_workloads.py
@@ -95,7 +95,11 @@ def wall_clock_time_in_msecs(
             kwargs["column_format"] = "llr|rr|r|r"
 
         return dataframe_to_table(
-            df, table_format, wrap_table, wrap_landscape=True, **kwargs
+            df,
+            table_format,
+            wrap_table=wrap_table,
+            wrap_landscape=True,
+            **kwargs
         )