Expand Up @@ -53,15 +53,15 @@ def __init__(*args, **kw): args = args[1:] if len(args) > 1: raise TypeError('expected at most 1 arguments, got %d' % len(args)) def remove(wr, selfref=ref(self)): def remove(wr, selfref=ref(self), _atomic_removal=_remove_dead_weakref): self = selfref() if self is not None: if self._iterating: self._pending_removals.append(wr.key) else: # Atomic removal is necessary since this function # can be called asynchronously by the GC _remove_dead_weakref(self.data, wr.key) _atomic_removal(self.data, wr.key) self._remove = remove # A list of keys to be removed self._pending_removals = [] Expand Down