diff --git a/spacy/kb/candidate.py b/spacy/kb/candidate.py index 663b68168..b32b72dd8 100644 --- a/spacy/kb/candidate.py +++ b/spacy/kb/candidate.py @@ -76,7 +76,7 @@ class InMemoryCandidate(Candidate): mention: str, entity_vector: List[float], prior_prob: float, - entity_freq: int + entity_freq: int, ): """ hash_to_str (Callable[[int], str]): Callable retrieving entity name from provided entity/vocab hash. diff --git a/spacy/tests/serialize/test_serialize_kb.py b/spacy/tests/serialize/test_serialize_kb.py index 860b9f8c7..3d75862c6 100644 --- a/spacy/tests/serialize/test_serialize_kb.py +++ b/spacy/tests/serialize/test_serialize_kb.py @@ -63,7 +63,9 @@ def _check_kb(kb): assert alias_string not in kb.get_alias_strings() # check candidates & probabilities - candidates = sorted(kb._get_alias_candidates("double07"), key=lambda x: x.entity_id_str) + candidates = sorted( + kb._get_alias_candidates("double07"), key=lambda x: x.entity_id_str + ) assert len(candidates) == 2 assert candidates[0].entity_id_str == "Q007"