Optimize summation convention for terms

tschijnmo · Apr 4, 2017 · a690cf0 · a690cf0
1 parent a63ed7f
commit a690cf0
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/drudge/term.py b/drudge/term.py
@@ -1458,6 +1458,9 @@ def replace_cb(_, *curr_indices):
     new_sums = []
     for symb, use in use_tally.items():
 
+        if symb in existing_dumms:
+            continue
+
         if use[0] != 2 and use[0] + use[1] != 2:
             # No chance to be an Einstein summation.
             continue
@@ -1484,8 +1487,7 @@ def replace_cb(_, *curr_indices):
             continue
 
         # Now we have an Einstein summation.
-        if symb not in existing_dumms:
-            new_sums.append((symb, range_))
+        new_sums.append((symb, range_))
         continue
 
     # Make summation from Einstein convention deterministic.