Make expire fast for completed units (#440)

Co-authored-by: Jeremy Francis Reizenstein <[email protected]>
facebookresearch · Apr 21, 2021 · 87b96a0 · 87b96a0
1 parent 2067a75
commit 87b96a0
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/mephisto/abstractions/providers/mturk/mturk_unit.py b/mephisto/abstractions/providers/mturk/mturk_unit.py
@@ -210,7 +210,10 @@ def expire(self) -> float:
         otherwise just return the maximum assignment duration
         """
         delay = 0
-        if self.get_status() == AssignmentState.ASSIGNED:
+        status = self.get_status()
+        if status in [AssignmentState.EXPIRED, AssignmentState.COMPLETED]:
+            return delay
+        if status == AssignmentState.ASSIGNED:
             # The assignment is currently being worked on,
             # so we will set the wait time to be the
             # amount of time we granted for working on this assignment