The key needs to be a type not an instance

quantumlib · Mar 3, 2025 · f2a7607 · f2a7607
1 parent c432d4f
commit f2a7607
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cirq-google/cirq_google/engine/calibration_to_noise_properties.py b/cirq-google/cirq_google/engine/calibration_to_noise_properties.py
@@ -54,12 +54,12 @@
 }
 GATE_PREFIX_PAIRS: Dict[Type['cirq.Gate'], str] = {
     cg_ops.SycamoreGate: 'two_qubit_parallel_sycamore_gate',
-    ops.CZ: 'two_qubit_parallel_cz_gate',
+    ops.CZPowGate: 'two_qubit_parallel_cz_gate',
     ops.ISwapPowGate: 'two_qubit_parallel_sqrt_iswap_gate',
 }
 GATE_ZPHASE_CODE_PAIRS: Dict[Type['cirq.Gate'], str] = {
     cg_ops.SycamoreGate: 'syc',
-    ops.CZ: 'cz',
+    ops.CZPowGate: 'cz',
     ops.ISwapPowGate: 'sqrt_iswap',
 }