Add the ability to reject submissions with AutoReviewPoller

IndicoDataSolutions · Jan 21, 2025 · 6243fe5 · 6243fe5
1 parent db6081f
commit 6243fe5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/examples/poll_auto_review.py b/examples/poll_auto_review.py
@@ -30,7 +30,8 @@ async def auto_review(
 
     return AutoReviewed(
         changes=predictions.to_changes(result),
-        stp=True,  # Defaults to `False` and may be omitted.
+        reject=False,  # Defaults to `False` and may be omitted.
+        stp=False,  # Defaults to `False` and may be omitted.
     )
 
 

diff --git a/indico_toolkit/polling/autoreview.py b/indico_toolkit/polling/autoreview.py
@@ -31,6 +31,7 @@
 @dataclass
 class AutoReviewed:
     changes: "dict[str, Any] | list[dict[str, Any]]"
+    reject: bool = False
     stp: bool = False
 
 
@@ -170,6 +171,7 @@ async def _worker(self, submission_id: "SubmissionId") -> None:
             SubmitReview(
                 submission_id,
                 changes=auto_reviewed.changes,
+                rejected=auto_reviewed.reject,
                 force_complete=auto_reviewed.stp,
             )
         )