Merge pull request #76 from TrueLearnAI/fix-74

fix: Title wrap to prevent offscreen topic labels

Author: KD-7

truelearn/tests/baseline_files/testbubbleplotter/test_bubble_word_overflow.png

@@ -308,6 +308,30 @@ def test_top_n(self, resources):
         plotter.plot(resources[0], top_n=3)
         return plotter
 
+    def test_bubble_word_overflow(self, resources):
+        plotter = visualisations.BubblePlotter()
+        plotter.plot(
+            resources[0],
+            topics=[
+                "Posterior probability",
+                "Algorithm",
+                "Machine learning",
+                "Variance",
+                "Exponential family",
+                "Factor graph",
+                "Rejection sampling",
+                "Probability",
+                "Topic model",
+                "Expected value",
+                "Bayesian inference",
+                "Markov chain",
+                "Importance sampling",
+                "Normalizing constant",
+                "Hyperplane",
+            ],
+        )
+        return plotter
+
 
 @file_comparison(plotter_type="plotly")
 class TestDotPlotter:

truelearn/tests/baseline_files/testbubbleplotter/test_default.png

@@ -2,7 +2,7 @@
 from typing_extensions import Self
 
 import circlify
-from matplotlib import cm, colors, patches
+from matplotlib import cm, colors, colormaps, patches
 import matplotlib.pyplot as plt
 
 from truelearn.models import Knowledge
@@ -85,7 +85,7 @@ def plot(
         plt.xlim(-lim, lim)
         plt.ylim(-lim, lim)
 
-        cmap = cm.get_cmap("Greens_r")
+        cmap = colormaps["Greens_r"]
 
         # Normalize data range to colormap range
         norm = colors.Normalize(vmin=min(variances) - 0.05, vmax=max(variances) + 0.05)
@@ -104,7 +104,10 @@ def plot(
                     )
                 )
                 plt.annotate(
-                    titles[len(titles) - 1 - i], (x, y), va="center", ha="center"
+                    titles[len(titles) - 1 - i].replace(" ", "\n"),
+                    (x, y),
+                    va="center",
+                    ha="center",
                 )
 
         # set up the colorbar on the right