use zip (#37)

Qiskit · Oct 10, 2024 · eaeab61 · eaeab61
1 parent ed5480a
commit eaeab61
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/qiskit_addon_dice_solver/dice_solver.py b/qiskit_addon_dice_solver/dice_solver.py
@@ -595,13 +595,12 @@ def _construct_ci_vec_from_addresses_amplitudes(
 
     addr_map = {uni_addr: i for i, uni_addr in enumerate(uniques)}
 
-    for idx, address in enumerate(addresses):
-        address_a = address[0]
-        address_b = address[1]
-
+    for amp, address in zip(amps, addresses):
+        address_a, address_b = address
         i = addr_map[address_a]
         j = addr_map[address_b]
-        ci_vec[i, j] = amps[idx]
+
+        ci_vec[i, j] = amp
 
         addresses_a[i] = uniques[i]
         addresses_b[j] = uniques[j]