[Shogi] Randomly initialize the current_player variable. (#1298)

Author: KazukiOhta

pgx/shogi.py

@@ -50,9 +50,14 @@ def __init__(self):
         self._game = Game()
 
     def _init(self, key: PRNGKey) -> State:
-        state = State()
-        player_order = jnp.array([[0, 1], [1, 0]])[jax.random.bernoulli(key).astype(jnp.int32)]
-        return state.replace(_player_order=player_order)  # type: ignore
+        x = GameState()
+        _player_order = jnp.array([[0, 1], [1, 0]])[jax.random.bernoulli(key).astype(jnp.int32)]
+        state = State(  # type: ignore
+            current_player=_player_order[x.color],
+            _player_order=_player_order,
+            _x=x,
+        )
+        return state
 
     def _step(self, state: core.State, action: Array, key) -> State:
         del key
@@ -83,4 +88,4 @@ def version(self) -> str:
 
     @property
     def num_players(self) -> int:
-        return 2
+        return 2