Update.

[picoclvr.git] / escape.py

diff --git a/escape.py b/escape.py
index 7596bea..5f34cd1 100755 (executable)
--- a/escape.py
+++ b/escape.py
@@ -242,7 +242,7 @@ def episodes2str(
         def status_bar(a, r, lr=None):
             a, r = a.item(), r.item()
             sb_a = "ISNEW"[a] if a >= 0 and a < 5 else "?"
-            sb_r = "- +"[r + 1] if r in {-1, 0, 1} else "?"
+            sb_r = "- +U"[r + 1] if r in {-1, 0, 1, 2} else "?"
             if lr is None:
                 sb_lr = ""
             else: