formatting and remove assertion

Author: tcochran-quera

src/bloqade/cirq_utils/noise/model.py

@@ -57,8 +57,8 @@ class GeminiNoiseModelABC(cirq.NoiseModel, MoveNoiseModelABC):
 
     def __post_init__(self):
         if (
-            self.cz_paired_correlated_rates is None
-            and self.cz_paired_error_probabilities is None
+                self.cz_paired_correlated_rates is None
+                and self.cz_paired_error_probabilities is None
         ):
             # NOTE: no input, set to default value; weird setattr for frozen dataclass
             object.__setattr__(
@@ -248,7 +248,8 @@ def noisy_moment(self, moment, system_qubits):
         # Check if the moment contains 1-qubit gates or 2-qubit gates
         elif len(moment.operations[0].qubits) == 1:
             if (isinstance(moment.operations[0].gate, cirq.ResetChannel)) or (
-            cirq.is_measurement(moment.operations[0])) or (isinstance(moment.operations[0].gate, cirq.BitFlipChannel)):
+                    cirq.is_measurement(moment.operations[0])) or (
+            isinstance(moment.operations[0].gate, cirq.BitFlipChannel)):
                 move_noise_ops = []
                 gate_noise_ops = []
             else:
@@ -354,8 +355,8 @@ def count_remaining_cz_moments(moments_2q):
         pm = 2 * self.sitter_pauli_rates[0]
         ps = 2 * self.cz_unpaired_pauli_rates[0]
 
-        #probability of a bitflip error for a sitting, unpaired qubit during a move/cz/move cycle.
-        heuristic_1step_bitflip_error: float = 2 * pm * (1 - ps) * (1- pm) + (1 - pm)**2 * ps + pm**2 * ps
+        # probability of a bitflip error for a sitting, unpaired qubit during a move/cz/move cycle.
+        heuristic_1step_bitflip_error: float = 2 * pm * (1 - ps) * (1 - pm) + (1 - pm) ** 2 * ps + pm ** 2 * ps
 
         for idx, moment in enumerate(moments_1q):
             interleaved_moments.append(moment)
@@ -412,7 +413,8 @@ def noisy_moment(self, moment, system_qubits):
         # Check if the moment contains 1-qubit gates or 2-qubit gates
         elif len(moment.operations[0].qubits) == 1:
             if (isinstance(moment.operations[0].gate, cirq.ResetChannel)) or (
-            cirq.is_measurement(moment.operations[0])) or (isinstance(moment.operations[0].gate, cirq.BitFlipChannel)):
+                    cirq.is_measurement(moment.operations[0])) or (
+            isinstance(moment.operations[0].gate, cirq.BitFlipChannel)):
                 gate_noise_ops = []
             else:
                 gate_noise_ops, _ = self._single_qubit_moment_noise_ops(

test/cirq_utils/noise/test_noise_models.py

@@ -14,7 +14,7 @@
 )
 
 
-def create_ghz_circuit(qubits, measurements:bool=False):
+def create_ghz_circuit(qubits, measurements: bool = False):
     n = len(qubits)
     circuit = cirq.Circuit()
 
@@ -38,23 +38,23 @@ def create_ghz_circuit(qubits, measurements:bool=False):
 @pytest.mark.parametrize(
     "model,qubits,measurements",
     [
-        (GeminiOneZoneNoiseModel(), None,False),
+        (GeminiOneZoneNoiseModel(), None, False),
         (
-            GeminiOneZoneNoiseModelConflictGraphMoves(),
-            cirq.GridQubit.rect(rows=1, cols=2),
-            False
+                GeminiOneZoneNoiseModelConflictGraphMoves(),
+                cirq.GridQubit.rect(rows=1, cols=2),
+                False
         ),
         (GeminiTwoZoneNoiseModel(), None, False),
         (GeminiOneZoneNoiseModel(), None, True),
         (
-            GeminiOneZoneNoiseModelConflictGraphMoves(),
-            cirq.GridQubit.rect(rows=1, cols=2),
-            True
+                GeminiOneZoneNoiseModelConflictGraphMoves(),
+                cirq.GridQubit.rect(rows=1, cols=2),
+                True
         ),
         (GeminiTwoZoneNoiseModel(), None, True),
     ],
 )
-def test_simple_model(model: cirq.NoiseModel, qubits, measurements:bool):
+def test_simple_model(model: cirq.NoiseModel, qubits, measurements: bool):
     if qubits is None:
         qubits = cirq.LineQubit.range(2)
 

test/cirq_utils/test_parallelize.py

@@ -31,6 +31,7 @@ def test1():
     circuit2 = parallelize(circuit)
     assert len(circuit2.moments) == 7
 
+
 def test_measurement_and_reset():
     qubits = cirq.LineQubit.range(4)
     circuit = cirq.Circuit(
@@ -52,14 +53,13 @@ def test_measurement_and_reset():
 
     parallelized_circuit = parallelize(circuit)
 
-    assert len(parallelized_circuit.moments) == 13
-
-    #this circuit should deterministically return all qubits to |0>
-    #let's check:
+    # this circuit should deterministically return all qubits to |0>
+    # let's check:
     simulator = cirq.Simulator()
-    for _ in range(20): #one in a million chance we miss an error
+    for _ in range(20):  # one in a million chance we miss an error
         state_vector = simulator.simulate(parallelized_circuit).state_vector()
-        assert np.all(np.isclose(np.abs(state_vector), np.concatenate((np.array([1]),np.zeros(2**4-1)))))
+        assert np.all(np.isclose(np.abs(state_vector), np.concatenate((np.array([1]), np.zeros(2 ** 4 - 1)))))
+
 
 def test_nonunitary_error_gate():
     qubits = cirq.LineQubit.range(2)
@@ -74,6 +74,7 @@ def test_nonunitary_error_gate():
 
     assert len(parallelized_circuit.moments) == 7
 
+
 RNG_STATE = np.random.RandomState(1902833)