Skip to content

Commit f2f6548

Browse files
committed
fix: don't overwrite last played
1 parent 1825f04 commit f2f6548

1 file changed

Lines changed: 10 additions & 11 deletions

File tree

  • kaggle_environments/envs/open_spiel_env/games/go/visualizer/v2/src/transformers

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

Lines changed: 10 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ export const goTransformer = (environment: any): GoStep[] => {
5353
const goSteps: GoStep[] = [];
5454

5555
const firstStep = goReplay.steps[0];
56-
const firstStepPlayers = [0, 1].map(
56+
const extraStepPlayers = [0, 1].map(
5757
(index): GoPlayer => ({
5858
id: index,
5959
name: environment.info.TeamNames[index],
@@ -68,7 +68,7 @@ export const goTransformer = (environment: any): GoStep[] => {
6868

6969
goSteps.push({
7070
step: goSteps.length,
71-
players: firstStepPlayers,
71+
players: extraStepPlayers,
7272
boardState: parseBoardState(firstStep[0].observation.observationString),
7373
isTerminal: false,
7474
winner: null,
@@ -97,16 +97,15 @@ export const goTransformer = (environment: any): GoStep[] => {
9797
isTerminal: step[0].observation.isTerminal,
9898
winner: null,
9999
});
100-
}
101-
102-
const endStep = Object.assign({}, goSteps.at(-1));
103-
endStep.step += 1;
104-
endStep.players[0].isTurn = false;
105-
endStep.players[1].isTurn = false;
106-
endStep.winner = deriveWinner(goReplay.steps[goReplay.steps.length - 1]);
107-
goSteps.push(endStep);
100+
};
108101

109-
console.log(goSteps);
102+
goSteps.push({
103+
step: goSteps.length,
104+
players: extraStepPlayers,
105+
boardState: goSteps[goSteps.length -1].boardState,
106+
isTerminal: goSteps[goSteps.length -1].isTerminal,
107+
winner: deriveWinner(goReplay.steps[goReplay.steps.length - 1]),
108+
});
110109

111110
return goSteps;
112111
};

0 commit comments

Comments
 (0)