fix: Fix KeyError by no element traveler

Fixes HOYO-BUDDY-XR
seriaati · Jan 10, 2025 · 021a7b0 · 021a7b0
1 parent 9d8dfce
commit 021a7b0
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/hoyo_buddy/ui/hoyo/characters.py b/hoyo_buddy/ui/hoyo/characters.py
@@ -577,7 +577,11 @@ async def start(self, i: Interaction) -> None:
 
             # Find traveler element and add 1 to the element char count
             for character in self.gi_characters:
-                if character.id in TRAVELER_IDS:
+                if (
+                    character.id in TRAVELER_IDS
+                    and character.element.lower()
+                    in self.element_char_counts  # Prevent KeyError caused by 'None' element traveler
+                ):
                     self.element_char_counts[character.element.lower()] += 1
                     break