Use defaultdict in bysize.

jaraco · Oct 20, 2022 · 8c0082d · 8c0082d
1 parent 7638e3f
commit 8c0082d
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/inflect/__init__.py b/inflect/__init__.py
@@ -55,6 +55,7 @@
 import ast
 import re
 import functools
+import collections
 import contextlib
 from typing import (
     Dict,
@@ -148,12 +149,10 @@ def bysize(words: Iterable[str]) -> Dict[int, set]:
     >>> ret[5]
     {'horse'}
     """
-    ret: Dict[int, set] = {}
+    res: Dict[int, set] = collections.defaultdict(set)
     for w in words:
-        if len(w) not in ret:
-            ret[len(w)] = set()
-        ret[len(w)].add(w)
-    return ret
+        res[len(w)].add(w)
+    return res
 
 
 def make_pl_si_lists(