Update src/qibo/backends/numpy.py

qiboteam · Jun 19, 2024 · eaac905 · eaac905
1 parent 9bb04fb
commit eaac905
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/src/qibo/backends/numpy.py b/src/qibo/backends/numpy.py
@@ -400,10 +400,7 @@ def execute_circuit(self, circuit, initial_state=None, nshots=1000):
                 return self.execute_circuit(initial_state + circuit, None, nshots)
         elif initial_state is not None:
             initial_state = self.cast(initial_state)
-            if circuit.density_matrix:
-                valid_shape = 2 * (2**circuit.nqubits,)
-            else:
-                valid_shape = (2**circuit.nqubits,)
+            valid_shape = 2 * (2**circuit.nqubits,) if circuit.density_matrix else (2**circuit.nqubits,)
             if tuple(initial_state.shape) != valid_shape:
                 raise_error(
                     ValueError,