diff --git a/hivemind/moe/client/beam_search.py b/hivemind/moe/client/beam_search.py index 4baef06a7..f8b357c7b 100644 --- a/hivemind/moe/client/beam_search.py +++ b/hivemind/moe/client/beam_search.py @@ -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