File tree Expand file tree Collapse file tree 1 file changed +6
-5
lines changed Expand file tree Collapse file tree 1 file changed +6
-5
lines changed Original file line number Diff line number Diff line change @@ -143,15 +143,16 @@ def process_op(old_op: Op):
143
143
return Op .product (new_ops )
144
144
145
145
# 1-e terms
146
- for p , q in itertools .product (range (norbs ), repeat = 2 ):
146
+ pairs = np .argwhere (h1e != 0 )
147
+ for p , q in pairs :
147
148
op = process_op (a_dag_ops [p ] * a_ops [q ])
148
149
ham_terms .append (op * h1e [p , q ])
149
150
150
151
# 2-e terms.
151
- for q , s in itertools . product ( range ( norbs ), repeat = 2 ):
152
- for p ,r in itertools . product ( range ( q ), range ( s )) :
153
- op = process_op (Op .product ([a_dag_ops [p ], a_dag_ops [q ], a_ops [r ], a_ops [s ]]))
154
- ham_terms .append (op * h2e [p , q , r , s ])
152
+ pairs = np . argwhere ( h2e != 0 )
153
+ for p , q , r , s in pairs :
154
+ op = process_op (Op .product ([a_dag_ops [p ], a_dag_ops [q ], a_ops [r ], a_ops [s ]]))
155
+ ham_terms .append (op * h2e [p , q , r , s ])
155
156
156
157
157
158
basis = []
You can’t perform that action at this time.
0 commit comments