feat: start and end steps (#820)

Author: jim-at-stink

kaggle_environments/envs/open_spiel_env/games/go/visualizer/v2/src/transformers/goTransformer.ts

@@ -52,11 +52,33 @@ export const goTransformer = (environment: any): GoStep[] => {
   const goReplay = environment as GoReplay;
   const goSteps: GoStep[] = [];
 
+  const firstStep = goReplay.steps[0];
+  const extraStepPlayers = [0, 1].map(
+    (index): GoPlayer => ({
+      id: index,
+      name: environment.info.TeamNames[index],
+      thumbnail: '',
+      isTurn: false,
+      actionDisplayText: '',
+      thoughts: '',
+      reward: null,
+      generateReturns: null,
+    })
+  );
+
+  goSteps.push({
+    step: goSteps.length,
+    players: extraStepPlayers,
+    boardState: parseBoardState(firstStep[0].observation.observationString),
+    isTerminal: false,
+    winner: null,
+  });
+
   for (const step of goReplay.steps) {
     if (step.some((p) => p.action?.actionString) === false) continue;
 
-    const stepPlayers: GoPlayer[] = step.map((player, index): GoPlayer => {
-      return {
+    const stepPlayers: GoPlayer[] = step.map(
+      (player, index): GoPlayer => ({
         id: index,
         name: environment.info.TeamNames[index],
         thumbnail: '',
@@ -65,17 +87,25 @@ export const goTransformer = (environment: any): GoStep[] => {
         thoughts: parseThoughts(player.action),
         reward: player.reward,
         generateReturns: player.action?.generate_returns ?? null,
-      };
-    });
+      })
+    );
 
     goSteps.push({
       step: goSteps.length,
       players: stepPlayers,
       boardState: parseBoardState(step[0].observation.observationString),
       isTerminal: step[0].observation.isTerminal,
-      winner: deriveWinner(step),
+      winner: null,
     });
   }
 
+  goSteps.push({
+    step: goSteps.length,
+    players: extraStepPlayers,
+    boardState: goSteps[goSteps.length - 1].boardState,
+    isTerminal: goSteps[goSteps.length - 1].isTerminal,
+    winner: deriveWinner(goReplay.steps[goReplay.steps.length - 1]),
+  });
+
   return goSteps;
 };

kaggle_environments/envs/open_spiel_env/games/go/visualizer/v2/src/utils/getStepLabel.ts

@@ -2,7 +2,16 @@ import { BaseGameStep } from '@kaggle-environments/core';
 
 export function getStepLabel(step: BaseGameStep) {
   const player = step.players.find((p) => p.isTurn);
+
+  if (player === undefined && step.step === 0) {
+    const black = step.players.at(0)?.name ?? 'Black';
+    const white = step.players.at(1)?.name ?? 'White';
+    return `${black} vs. ${white}`;
+  }
+
+  if (player === undefined) return '';
+
   const color = player?.id ? 'white' : 'black';
   const move = player?.actionDisplayText?.toUpperCase() ?? '';
-  return `Plays a ${color} stone on ${move}`;
+  return `Played the ${color} stone on ${move}`;
 }