Adds small optimization of subclasses

blindspot-ai · May 13, 2019 · 44f33e6 · 44f33e6
1 parent 8fe1532
commit 44f33e6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ftoolz/unsafe/reflection.py b/ftoolz/unsafe/reflection.py
@@ -161,8 +161,8 @@ def subclasses(clz: Type[T], package: str) -> Set[Type[T]]:
     """
 
     def rec(cls: Type[T]) -> Set[Type[T]]:
-        return set(cls.__subclasses__()) \
-            .union([s for c in cls.__subclasses__() for s in rec(c)])
+        sub = set(cls.__subclasses__())
+        return sub.union(s for c in sub for s in rec(c))
 
     import_all(package)
     return rec(clz)