fix: schemas

app/schemas.py

@@ -184,7 +184,6 @@ class ElectionCreate(ElectionBase):
 
     @validator("grades")
     def all_grades_have_unique_values_and_names(cls, grades: list[GradeBase]):
-        grades = [grade for grade in grades if grade.id is not None]
 
         values = [g.value for g in grades]
         if len(set(values)) != len(grades):
@@ -246,7 +245,9 @@ class ElectionUpdate(ElectionBase):
         return value
 
     @validator("grades")
-    def all_grades_have_unique_values_ids_and_names(cls, grades: list[GradeUpdate]):
+    def all_grades_have_unique_values_and_names(cls, grades: list[GradeBase]):
+        grades = [grade for grade in grades if grade.id is not None]
+
         values = [g.value for g in grades]
         if len(set(values)) != len(grades):
             raise ArgumentsSchemaError("At least two grades have the same value")