Cleanup flatten code

cjdrake · Feb 11, 2015 · 874aad0 · 874aad0
1 parent 538d33b
commit 874aad0
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pyeda/boolalg/expr.py b/pyeda/boolalg/expr.py
@@ -1506,9 +1506,9 @@ def _flatten(self, op):
         if ex.depth < 2 or isinstance(ex, op.get_dual()):
             return ex
         else:
-            args = {x._lits for x in ex.xs}
-            prod = {frozenset(t) for t in itertools.product(*args)}
-            return op.get_dual()(*[op(*t) for t in prod]).simplify()
+            terms = {x._lits for x in ex.xs}
+            xs = {op(*t) for t in itertools.product(*terms)}
+            return op.get_dual()(*xs).simplify()._absorb()
 
     # FlattenedExpression
     @cached_property