- sb = sb + thin_vert + +("- +"[lr + 1] if lr in {-1, 0, 1} else "?")
- return sb + " " * (states.size(-1) - len(sb))
-
- result += (
- vert
- + vert.join([status_bar(a, r) for a, r in zip(actions[n], rewards[n])])
- + vert
- + "\n"
- )
+ sb_r = sb_r + "/" + ("- +"[lr + 1] if lr in {-1, 0, 1} else "?")
+ return sb_a + " " * (states.size(-1) - len(sb_a) - len(sb_r)) + sb_r
+
+ if lookahead_rewards is None:
+ result += (
+ vert
+ + vert.join([status_bar(a, r) for a, r in zip(actions[n], rewards[n])])
+ + vert
+ + "\n"
+ )
+ else:
+ result += (
+ vert
+ + vert.join(
+ [
+ status_bar(a, r, lr)
+ for a, r, lr in zip(
+ actions[n], rewards[n], lookahead_rewards[n]
+ )
+ ]
+ )
+ + vert
+ + "\n"
+ )