Add c_label parameter to get_scatter2d and adjust margin settings

Author: zpeng

viz/viz.py

@@ -171,6 +171,7 @@ def get_scatter2d(
     c_key: str,
     x_label: Optional[str] = None,
     y_label: Optional[str] = None,
+    c_label: Optional[str] = None,
     uirevision: str = "no_change",
     colormap: str = "Jet",
     margin: Optional[Dict[str, int]] = None,
@@ -200,15 +201,16 @@ def get_scatter2d(
     linewidth = kwargs.get("linewidth", 0)
 
     if margin is None:
-        margin = {"l": 80, "r": 40, "b": 40, "t": 60}
+        margin = {"l": 70, "r": 60, "b": 60, "t": 60}
 
     if x_label is None:
         x_label = x_key
 
     if y_label is None:
         y_label = y_key
 
-    c_label = kwargs.get("c_label", c_key)
+    if c_label is None:
+        c_label = c_key
     c_type = kwargs.get("c_type", "numerical")
 
     if c_type == "numerical":
@@ -226,7 +228,7 @@ def get_scatter2d(
                         "colorscale": colormap,
                         "opacity": 0.8,
                         "colorbar": {
-                            "title": c_label,
+                            "title": {"text": c_label, "side": "right"},
                         },
                         "line": {
                             "color": "#FFFFFF",