improved

mate-academy · Jan 10, 2025 · 3abadd7 · 3abadd7
1 parent 253fa7c
commit 3abadd7
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/app/main.py b/app/main.py
@@ -29,11 +29,14 @@ def __init__(
         self.health = len(self.decks)
 
     def get_deck(self, row: int, column: int) -> Deck:
-        return [
+        if result :=  [
             deck
             for deck in self.decks
             if deck.row == row and deck.column == column
-        ][0]
+        ][0]:
+            return result
+        else:
+            raise IndexError
 
     def fire(self, row: int, column: int) -> None:
         if deck := self.get_deck(row, column):