Fix mapping bug

Qiskit · Sep 28, 2021 · 768d590 · 768d590
1 parent d545dc7
commit 768d590
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mthree/mitigation.py b/mthree/mitigation.py
@@ -401,7 +401,7 @@ def apply_correction(self, counts, qubits, distance=None,
 
         if isinstance(qubits, dict):
             # If a mapping was given for qubits
-            qubits = list(qubits)
+            qubits = [list(qubits)]
         elif not any(isinstance(qq, (list, tuple, np.ndarray, dict)) for qq in qubits):
             qubits = [qubits]*len(counts)
         else: