most_similar code format

Xeadriel · Nov 7, 2022 · f7ee0a8 · f7ee0a8
1 parent 1c53d2e
commit f7ee0a8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/firelang/models/_fireword.py b/firelang/models/_fireword.py
@@ -141,7 +141,10 @@ def field(
 
     @torch.no_grad()
     def most_similar(
-        self, word: str, k: int = 10, p: float = 0.3,
+        self,
+        word: str,
+        k: int = 10,
+        p: float = 0.3,
     ) -> List[Tuple[str, float]]:
         """Return the most similar `k` words to `word`, as well as the (frequency-adjusted) similarity scores.