diff --git a/u5/kindergarten_garden.py b/u5/kindergarten_garden.py index 191fe1b..0bfc937 100644 --- a/u5/kindergarten_garden.py +++ b/u5/kindergarten_garden.py @@ -3,8 +3,7 @@ class NoSuchStudentException(Exception): DEFAULT_STUDENTS = [ "Alena", "Boris", "Cyril", "Daniel", "Eva", "FrantiĊĦek", - "Gitka", "Helena", "Ivana", "Jozef", "Karol", "Ladislav", -] + "Gitka", "Helena", "Ivana", "Jozef", "Karol", "Ladislav",] PLANTS = {"G": "Trava", "C": "Datelina", "R": "Redkovky", "V": "Fialky"} class Garden(): @@ -22,4 +21,4 @@ class Garden(): raise NoSuchStudentException(f"Student {student} does not exist") start = self.students.index(student) * 2 - return [PLANTS[code] for row in self.rows for code in row[start:start + 2]] + return [PLANTS[code] for row in self.rows for code in row[start:start + 2]]