From 0ebbd9ce24eb49baf5ffaa637d3a44e6c1e269e9 Mon Sep 17 00:00:00 2001
From: AJ Fontaine <fontbane@gmail.com>
Date: Mon, 16 Dec 2024 17:51:45 -0500
Subject: [PATCH] Check game mode for wave 20 trainer evo ban

---
 src/data/pokemon-species.ts | 3 ---
 src/data/trainer-config.ts  | 3 +++
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/data/pokemon-species.ts b/src/data/pokemon-species.ts
index cfb5d2e23f3..09788e353cf 100644
--- a/src/data/pokemon-species.ts
+++ b/src/data/pokemon-species.ts
@@ -670,9 +670,6 @@ export default class PokemonSpecies extends PokemonSpeciesForm implements Locali
   }
 
   getTrainerSpeciesForLevel(level: number, allowEvolving: boolean = false, strength: PartyMemberStrength, currentWave: number = 0): Species {
-    if (currentWave === 20) { //Don't allow first gym leader wave 20 to evolve their mons
-      allowEvolving = false;
-    }
     return this.getSpeciesForLevel(level, allowEvolving, true, strength, currentWave);
   }
 
diff --git a/src/data/trainer-config.ts b/src/data/trainer-config.ts
index f473ae227bb..2d8c7a3cbc7 100644
--- a/src/data/trainer-config.ts
+++ b/src/data/trainer-config.ts
@@ -1170,6 +1170,9 @@ function getGymLeaderPartyTemplate(scene: BattleScene) {
 export function getRandomPartyMemberFunc(speciesPool: Species[], trainerSlot: TrainerSlot = TrainerSlot.TRAINER, ignoreEvolution: boolean = false, postProcess?: (enemyPokemon: EnemyPokemon) => void) {
   return (scene: BattleScene, level: number, strength: PartyMemberStrength) => {
     let species = Utils.randSeedItem(speciesPool);
+    if (scene.gameMode.isClassic && scene.currentBattle.waveIndex === 20) {
+      ignoreEvolution = true;
+    }
     if (!ignoreEvolution) {
       species = getPokemonSpecies(species).getTrainerSpeciesForLevel(level, true, strength, scene.currentBattle.waveIndex);
     }