fix: maia prediction checkmarks for varying maia versions

CSSLab · Dec 6, 2024 · 38ec884 · 38ec884
1 parent ca26d3a
commit 38ec884
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/components/Board/MovesContainer.tsx b/src/components/Board/MovesContainer.tsx
@@ -13,6 +13,7 @@ interface Props {
   highlightIndices?: number[]
   mobile?: boolean
   termination?: Termination
+  currentMaiaModel?: string
 }
 
 export const MovesContainer: React.FC<Props> = ({
@@ -21,6 +22,7 @@ export const MovesContainer: React.FC<Props> = ({
   highlightIndices,
   mobile = false,
   termination,
+  currentMaiaModel,
 }: Props) => {
   const controller = useContext(GameControllerContext)
   const { isMobile } = useContext(WindowSizeContext)
@@ -66,7 +68,8 @@ export const MovesContainer: React.FC<Props> = ({
         let predictedWhite, predictedBlack
 
         if ('maiaEvaluations' in game) {
-          const evals = game.maiaEvaluations['maia_kdd_1100']
+          const evals =
+            game.maiaEvaluations[currentMaiaModel ?? 'maia_kdd_1100']
 
           const prevEvals = evals?.[prevMoveIndex]
           const whiteEvals = evals?.[whiteMoveIndex]

diff --git a/src/pages/analysis/[...id].tsx b/src/pages/analysis/[...id].tsx
@@ -552,6 +552,7 @@ const Analysis: React.FC<Props> = ({
                 game={analyzedGame}
                 setCurrentMove={setCurrentMove}
                 termination={analyzedGame.termination}
+                currentMaiaModel={currentMaiaModel}
               />
             </div>
             <div className="flex-none">
@@ -653,6 +654,7 @@ const Analysis: React.FC<Props> = ({
                 game={analyzedGame}
                 setCurrentMove={setCurrentMove}
                 termination={analyzedGame.termination}
+                currentMaiaModel={currentMaiaModel}
               />
             </div>
             <div>