handle (2,1,p)-encoded single dvar

qiskit-community · Jun 9, 2022 · 5010949 · 5010949
1 parent b6a5e0e
commit 5010949
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/qrao/magic_rounding.py b/qrao/magic_rounding.py
@@ -352,7 +352,7 @@ def _sample_bases_weighted(self, q2vars, trace_values, vars_per_qubit):
                 basis_probs.append([ppp_mmm, ppm_mmp, mpm_pmp, pmm_mpp])
             elif vars_per_qubit == 2:
                 x = 0.5 * (1 - tv[dvars[0]])
-                z = 0.5 * (1 - tv[dvars[1]])
+                z = 0.5 * (1 - tv[dvars[1]]) if (len(dvars) > 1) else 0
                 # pp:   xi±   = .5(I ± 1/sqrt(2)( X + Z ))
                 # pm: X xi± X = .5(I ± 1/sqrt(2)( X - Z ))
                 # fmt: off