justheuristic 4 years ago
parent
commit
706fa16d86
1 changed files with 1 additions and 1 deletions
  1. 1 1
      hivemind/utils/mpfuture.py

+ 1 - 1
hivemind/utils/mpfuture.py

@@ -156,7 +156,7 @@ class MPFuture(base.Future, Generic[ResultType]):
                 if msg_type == MessageType.STATE_REQUEST:
                 if msg_type == MessageType.STATE_REQUEST:
                     future_state = None if future is None else future.__getstate__()
                     future_state = None if future is None else future.__getstate__()
                     use_lock, return_pipe = payload
                     use_lock, return_pipe = payload
-                    with nullcontext():
+                    with MPFuture._update_lock if use_lock else nullcontext():
                         return_pipe.send((uid, MessageType.STATE_RESPONSE, future_state))
                         return_pipe.send((uid, MessageType.STATE_RESPONSE, future_state))
 
 
                 elif msg_type == MessageType.STATE_RESPONSE:
                 elif msg_type == MessageType.STATE_RESPONSE: