Record the model's raw completion for each played move

arena.py

@@ -34,6 +34,13 @@ def choose(game: TetrisGame, player, protocol: str, seed: int):
     rng = random.Random(f"{seed}:{game.pieces}")
     i = ties[rng.randrange(len(ties))]
     game.apply(cands[i], vals[i], len(cands))
+    # the model's raw completion for the move it played (shown under the board; baselines have none)
+    game.last_raw = None
+    if hasattr(player, "complete"):
+        try:
+            game.last_raw = player.complete(protocol, cands[i].description)
+        except Exception:
+            game.last_raw = None
     return True