reset masks when resetting, because map size may have changed

Farama-Foundation · Dec 19, 2023 · b7225f1 · b7225f1
1 parent 50a912f
commit b7225f1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/tests/JNIGridnetClient.java b/src/tests/JNIGridnetClient.java
@@ -230,6 +230,7 @@ public Response reset(int player) throws Exception {
         ai2 = ai2.clone();
         ai2.reset();
         pgs = PhysicalGameState.load(mapPath, utt);
+        masks = new int[pgs.getHeight()][pgs.getWidth()][1+6+4+4+4+4+utt.getUnitTypes().size()+maxAttackRadius*maxAttackRadius];
         gs = new GameState(pgs, utt);
         if (partialObs) {
             player1gs = new PartiallyObservableGameState(gs, player);

diff --git a/src/tests/JNIGridnetClientSelfPlay.java b/src/tests/JNIGridnetClientSelfPlay.java
@@ -220,6 +220,11 @@ public String sendUTT() throws Exception {
 
     public void reset() throws Exception {
         pgs = PhysicalGameState.load(mapPath, utt);
+
+        for (int i = 0; i < numPlayers; i++) {
+            masks[i] = new int[pgs.getHeight()][pgs.getWidth()][1+6+4+4+4+4+utt.getUnitTypes().size()+maxAttackRadius*maxAttackRadius];
+        }
+
         gs = new GameState(pgs, utt);
         for (int i = 0; i < numPlayers; i++) {
             playergs[i] = gs;