Add method to programmatically determine whether a Pokemon is a Drago…

…n-type (#2)
pkmn · Aug 26, 2024 · 652db3e · 652db3e
1 parent d63b96b
commit 652db3e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/stats/src/classifier.ts b/stats/src/classifier.ts
@@ -255,7 +255,8 @@ function tag(gen: Generation, team: Array<PokemonSet<ID>>, stalliness: number, l
       pokemon.moves.some((m: ID) => ['block', 'meanlook', 'spiderweb'].includes(m))) {
       style.trappers++;
     }
-    if (style.dragons < 2 && DRAGONS.has(pokemon.species)) {
+    if (style.dragons < 2 && legacy ? DRAGONS.has(pokemon.species)
+      : gen.species.get(pokemon.species)?.types.includes('Dragon')) {
       style.dragons++;
     }
     if ((style.clearance < 2 && pokemon.ability === 'magicbounce') || moves.has('rapidspin')) {