quick bugfix in symbolic_mpo

shuaigroup · Nov 24, 2022 · bac8bed · bac8bed
1 parent 871caaa
commit bac8bed
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/renormalizer/mps/symbolic_mpo.py b/renormalizer/mps/symbolic_mpo.py
@@ -342,8 +342,8 @@ def _terms_to_table(model: Model, terms: List[Op], const: float):
 
     table = []
     factor_list = []
-
-    dummy_table_entry = [Op.identity(b.dof) for b in model.basis]
+    
+    dummy_table_entry = [Op.identity(b.dof) if not b.multi_dof else Op.identity(b.dof[0]) for b in model.basis]
     for op in terms:
         elem_ops, factor = op.split_elementary(model.dof_to_siteidx)
         table_entry = dummy_table_entry.copy()