dark

eval_viz_utils.py

@@ -20,6 +20,7 @@ logger = logging.getLogger(__name__)
 PROGRESS_COLOR = "#B20000"
 SUCCESS_COLOR = "#B20000"
 THEME_LIGHT = {"facecolor": "white", "text_color": "black", "spine_color": "#333333"}
+THEME_DARK = {"facecolor": "black", "text_color": "white", "spine_color": "#444444"}
 
 # Serif font (Palatino) for plots
 plt.rcParams["font.family"] = "serif"
@@ -27,6 +28,30 @@ plt.rcParams["font.serif"] = ["Palatino", "Palatino Linotype", "DejaVu Serif", "
 plt.rcParams["font.size"] = 11
 
 
+def wrap_title(text: str, max_chars_per_line: int = 48) -> str:
+    """Wrap a long title onto at most two lines, breaking at word boundaries."""
+    if not text or not str(text).strip():
+        return text
+    text = str(text).strip()
+    if len(text) <= max_chars_per_line:
+        return text
+    words = text.split()
+    line1, line2 = [], []
+    line1_len = 0
+    for w in words:
+        need = len(w) + (1 if line1 else 0)  # space before if not first
+        if line2:
+            line2.append(w)
+        elif line1_len + need <= max_chars_per_line:
+            line1.append(w)
+            line1_len += need
+        else:
+            line2.append(w)
+    if not line2:
+        return text
+    return " ".join(line1) + "\n" + " ".join(line2)
+
+
 def create_combined_progress_success_plot(
     progress_pred: np.ndarray,
     num_frames: int,
@@ -76,7 +101,7 @@ def create_combined_progress_success_plot(
     ax.plot(progress_pred, linewidth=2)
     ax.set_ylabel("Progress")
 
-    # Build title
+    # Build title (wrap long task text onto two lines)
     if title is None:
         title_parts = ["Progress"]
         if loss is not None:
@@ -84,7 +109,7 @@ def create_combined_progress_success_plot(
         if pearson is not None:
             title_parts.append(f"Pearson: {pearson:.2f}")
         title = ", ".join(title_parts)
-    fig.suptitle(title)
+    fig.suptitle(wrap_title(title))
 
     # Set y-limits and ticks (always continuous since discrete is converted before this function)
     ax.set_ylim(0, 1)
@@ -355,9 +380,9 @@ def create_progress_success_gif(
             head_dots.append(head_dot)
 
     if title and str(title).strip():
-        # Place title inside figure top margin (rect keeps axes below 0.88)
+        # Place title inside figure top margin; wrap long task text onto two lines
         fig.suptitle(
-            str(title).strip(),
+            wrap_title(str(title).strip()),
             fontsize=12,
             fontweight="bold",
             color=theme["text_color"],