Skip to content

Commit

Permalink
Merge branch 'master' into issue-526
Browse files Browse the repository at this point in the history
  • Loading branch information
ikmckenz authored Dec 15, 2022
2 parents c35b606 + 4c9c477 commit 7326141
Show file tree
Hide file tree
Showing 3 changed files with 5 additions and 3 deletions.
2 changes: 2 additions & 0 deletions hivemind/averaging/averager.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@
import multiprocessing as mp
import os
import random
import signal
import threading
import weakref
from dataclasses import asdict
Expand Down Expand Up @@ -329,6 +330,7 @@ def run_in_background(self, await_ready: bool = True, timeout: Optional[float] =
Starts averager in a background process. if await_ready, this method will wait until background dht
is ready to process incoming requests or for :timeout: seconds max.
"""
signal.signal(signal.SIGINT, signal.SIG_IGN)
self.start()
if await_ready:
self.wait_until_ready(timeout)
Expand Down
2 changes: 1 addition & 1 deletion hivemind/moe/client/beam_search.py
Original file line number Diff line number Diff line change
Expand Up @@ -171,7 +171,7 @@ def get_active_successors(
) -> Dict[ExpertPrefix, Dict[Coordinate, ExpertInfo]]:
"""
:param prefixes: a list of prefix for which to find active successor uids
:param grid_size: if specified, only return successors if ther are in range [0, grid_size)
:param grid_size: if specified, only return successors if they are in range [0, grid_size)
:param return_future: if False (default), find and return successors. Otherwise return MPFuture and fill later.
:returns: for every expert, return a dict{active_next_coordinate: (matching_expert_uid, matching_endpoint)}
:note: if a prefix is not found, get_active_successors will return an empty dictionary for that prefix
Expand Down
4 changes: 2 additions & 2 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,14 +13,14 @@
from setuptools.command.build_py import build_py
from setuptools.command.develop import develop

P2PD_VERSION = "v0.3.13"
P2PD_VERSION = "v0.3.14"

P2PD_SOURCE_URL = f"https://github.com/learning-at-home/go-libp2p-daemon/archive/refs/tags/{P2PD_VERSION}.tar.gz"
P2PD_BINARY_URL = f"https://github.com/learning-at-home/go-libp2p-daemon/releases/download/{P2PD_VERSION}/"

# The value is sha256 of the binary from the release page
EXECUTABLES = {
"p2pd": "c3bf9bffdf649ec295f515ed744882d7e197220846a21825e8f11c007f01dbc0",
"p2pd": "c9e7cde84ca53925622ad52fef50e41f8fd8cb258bbd69d4ab560beb672bb255",
}

here = os.path.abspath(os.path.dirname(__file__))
Expand Down

0 comments on commit 7326141

Please sign in to comment.