fix: prevent hangs in notification methods

x42005e1f · Oct 20, 2024 · 207ad17 · 207ad17
1 parent 019881a
commit 207ad17
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/aiologic/locks/condition.py b/src/aiologic/locks/condition.py
@@ -175,7 +175,7 @@ def notify(self, /, count=1, *, deadline=None):
                         notified += 1
 
                     try:
-                        waiters.remove(event)
+                        waiters.remove(token)
                     except ValueError:
                         pass
                 else:
@@ -204,7 +204,7 @@ def notify_all(self, /, *, deadline=None):
                         notified += 1
 
                     try:
-                        waiters.remove(event)
+                        waiters.remove(token)
                     except ValueError:
                         pass
                 else: