fixing broken tests from merging main

Author: weinbe58

src/bloqade/pyqrack/squin/op.py

@@ -18,7 +18,7 @@
     OperatorRuntime,
     ProjectorRuntime,
     OperatorRuntimeABC,
-    CliffordStringRuntime,
+    PauliStringRuntime,
 )
 
 
@@ -142,13 +142,13 @@ def u3(
         lam = frame.get(stmt.lam)
         return (U3Runtime(theta, phi, lam),)
 
-    @interp.impl(op.stmts.CliffordString)
+    @interp.impl(op.stmts.PauliString)
     def clifford_string(
         self,
         interp: PyQrackInterpreter,
         frame: interp.Frame,
-        stmt: op.stmts.CliffordString,
+        stmt: op.stmts.PauliString,
     ) -> tuple[OperatorRuntimeABC]:
         string = stmt.string
         ops = [OperatorRuntime(method_name=name.lower()) for name in stmt.string]
-        return (CliffordStringRuntime(string, ops),)
+        return (PauliStringRuntime(string, ops),)

src/bloqade/pyqrack/squin/runtime.py

@@ -309,14 +309,14 @@ def control_apply(self, *qubits: PyQrackQubit, adjoint: bool = False) -> None:
 
 
 @dataclass(frozen=True)
-class CliffordStringRuntime(NonBroadcastableOperatorRuntimeABC):
+class PauliStringRuntime(NonBroadcastableOperatorRuntimeABC):
     string: str
     ops: list[OperatorRuntime]
 
     def apply(self, *qubits: PyQrackQubit, adjoint: bool = False):
         if len(self.ops) != len(qubits):
             raise RuntimeError(
-                f"Cannot apply Clifford string {self.string} to {len(qubits)} qubits! Make sure the length matches."
+                f"Cannot apply Pauli string {self.string} to {len(qubits)} qubits! Make sure the length matches."
             )
 
         for i, op in enumerate(self.ops):

src/bloqade/squin/analysis/nsites/impls.py

@@ -1,6 +1,4 @@
-from typing import cast
-
-from kirin import ir, interp
+from kirin import interp
 
 from bloqade.squin import op
 
@@ -52,9 +50,7 @@ def control(
 
         if isinstance(op_sites, NumberSites):
             n_sites = op_sites.sites
-            n_controls_attr = stmt.get_attr_or_prop("n_controls")
-            n_controls = cast(ir.PyAttr[int], n_controls_attr).data
-            return (NumberSites(sites=n_sites + n_controls),)
+            return (NumberSites(sites=n_sites + stmt.n_controls),)
         else:
             return (NoSites(),)
 

src/bloqade/squin/op/__init__.py

@@ -87,12 +87,8 @@ def spin_p() -> types.Op: ...
 def u(theta: float, phi: float, lam: float) -> types.Op: ...
 
 
-@_wraps(stmts.CliffordString)
-def clifford_string(string: str) -> types.Op: ...
-
-
-@_wraps(stmts.CliffordString)
-def pauli_string(string: str) -> types.Op: ...
+@_wraps(stmts.PauliString)
+def pauli_string(*, string: str) -> types.Op: ...
 
 
 # stdlibs

test/pyqrack/test_squin.py

@@ -14,7 +14,9 @@ def new():
 
     new.print()
 
-    target = PyQrack(3)
+    target = PyQrack(
+        3, pyqrack_options={"isBinaryDecisionTree": False, "isStabilizerHybrid": True}
+    )
     result = target.run(new)
     assert isinstance(result, ilist.IList)
     assert isinstance(qubit := result[0], PyQrackQubit)
@@ -398,7 +400,7 @@ def test_clifford_str():
     @squin.kernel
     def main():
         q = squin.qubit.new(3)
-        cstr = squin.op.clifford_string(string="XXX")
+        cstr = squin.op.pauli_string(string="XXX")
         squin.qubit.apply(cstr, q)
         return squin.qubit.measure(q)