Update database.py

kevoreilly · May 24, 2024 · 219906a · 219906a
1 parent 44a6759
commit 219906a
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/lib/cuckoo/core/database.py b/lib/cuckoo/core/database.py
@@ -779,11 +779,7 @@ def set_status(self, task_id: int, status) -> Optional[Task]:
         @param status: status string
         @return: operation status
         """
-        task = False
-        try:
-            task = self.session.get(Task, task_id)
-        except PendingRollbackError:
-            self.session.rollback()
+        task = self.session.get(Task, task_id)
 
         if not task:
             return None