Remove bad type annotation

probcomp · Jul 18, 2024 · bd3bcb5 · bd3bcb5
1 parent cd2b3ea
commit bd3bcb5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hfppl/llms.py b/hfppl/llms.py
@@ -37,7 +37,7 @@ def __init__(self, lm):
 
         self.MAX_TOKEN_LENGTH = self.precompute_token_length_masks(lm)
 
-    def precompute_token_length_masks(self, lm) -> Dict[int, Set[int]]:
+    def precompute_token_length_masks(self, lm):
         """Precompute masks for tokens of different lengths.
 
         Each mask is a set of token ids that are of the given length or shorter."""