two flag bits for Steane.qec_tel

PECOS-packages · Dec 17, 2024 · 4c64865 · 4c64865
1 parent 0e8ae1d
commit 4c64865
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/python/quantum-pecos/src/pecos/qeclib/steane/steane_class.py b/python/quantum-pecos/src/pecos/qeclib/steane/steane_class.py
@@ -508,13 +508,14 @@ def qec_tel(
         aux: Steane,
         reject_x: Bit | None = None,
         reject_z: Bit | None = None,
-        flag_bit: Bit | None = None,
+        flag_bit_x: Bit | None = None,
+        flag_bit_z: Bit | None = None,
         rus_limit: int | None = None,
     ) -> Block:
         """Run a teleportation-based error correction cycle."""
         return Block(
-            self.qec_tel_x(aux, reject_x, rus_limit=rus_limit),
-            self.qec_tel_z(aux, reject_x, flag_bit, rus_limit),
+            self.qec_tel_x(aux, reject_x, flag_bit_x, rus_limit),
+            self.qec_tel_z(aux, reject_z, flag_bit_z, rus_limit),
         )
 
     def qec_tel_x(