[Shogi] Extract logic functions (#1269)

sotetsuk · Oct 31, 2024 · 08e1b93 · 08e1b93
1 parent e9d7c87
commit 08e1b93
Show file tree

Hide file tree

Showing 3 changed files with 95 additions and 93 deletions.
diff --git a/pgx/_src/dwg/shogi.py b/pgx/_src/dwg/shogi.py
@@ -7,7 +7,7 @@ def _make_shogi_dwg(dwg, state: ShogiState, config):  # noqa: C901
     if state._x.turn == 1:
         from pgx.shogi import _flip
 
-        state = _flip(state)
+        state = ShogiState(_x=_flip(state._x))  # type: ignore
     # fmt: off
     PIECES = ["歩", "香", "桂", "銀", "角", "飛", "金", "玉", "と", "成香", "成桂", "成銀", "馬", "龍"] * 2
     NUM_TO_CHAR = ["一", "二", "三", "四", "五", "六", "七", "八", "九", "十"]