Fixed plotting issue

src/components/quizbowl/plotting.py

@@ -169,7 +169,8 @@ def create_tossup_confidence_pyplot(
     logprobs = [o["logprob"] for o in run_outputs if o["logprob"] is not None]
     y_prob = [0] + [np.exp(v) for v in logprobs]
 
-    ax.plot(x, y_prob, "o-", color="#f2b150", label="Probability")
+    if prob_threshold is not None:
+        ax.plot(x, y_prob, "o-", color="#f2b150", label="Probability")
     ax.plot(x, y_conf, "o-", color="#4996de", label="Confidence")
     for o in run_outputs:
         if not o["buzz"]:
@@ -178,9 +179,10 @@ def create_tossup_confidence_pyplot(
         conf = o["confidence"]
         i = o["token_position"]
         ax.plot(i, conf, "o", color=color, markerfacecolor="none", markersize=12, markeredgewidth=2.5)
-        if o["logprob"] is not None:
+        if o["logprob"] is not None and prob_threshold is not None:
             prob = np.exp(o["logprob"])
-            ax.plot(i, prob, "o", color=color, markerfacecolor="none", markersize=12, markeredgewidth=2.5)
+            if prob > prob_threshold:
+                ax.plot(i, prob, "o", color=color, markerfacecolor="none", markersize=12, markeredgewidth=2.5)
         if i > len(tokens):
             print(f"1-indexed token index {i} is out of bounds for n_tokens: {len(tokens)}")
         ax.annotate(f"{tokens[i - 1]}", (i, conf), textcoords="offset points", xytext=(0, 10), ha="center")