add safe and unsafe to the enum in the validation table

mlcommons · Aug 12, 2024 · e299c55 · e299c55
1 parent 32c6ead
commit e299c55
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/backend/app/infrastructure/models/models.py b/backend/app/infrastructure/models/models.py
@@ -466,7 +466,9 @@ class Validation(Base):
     id = Column(Integer, primary_key=True)
     uid = Column(ForeignKey("users.id"), index=True)
     eid = Column(ForeignKey("examples.id"), nullable=False, index=True)
-    label = Column(Enum("flagged", "correct", "incorrect", "placeholder"))
+    label = Column(
+        Enum("flagged", "correct", "incorrect", "placeholder", "safe", "unsafe")
+    )
     mode = Column(Enum("user", "owner"))
     metadata_json = Column(Text)