finally fix this issue

qiboteam · Mar 14, 2024 · 07d28fa · 07d28fa
1 parent 05c1966
commit 07d28fa
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/src/qibo/backends/numpy.py b/src/qibo/backends/numpy.py
@@ -280,10 +280,11 @@ def _append_zeros(self, state, qubits, results):
         """Helper method for collapse."""
         for q, r in zip(qubits, results):
             state = self.np.expand_dims(state, q)
-            if r:
-                state = self.np.concatenate([self.np.zeros_like(state), state], q)
-            else:
-                state = self.np.concatenate([state, self.np.zeros_like(state)], q)
+            state = (
+                self.np.concatenate([self.np.zeros_like(state), state], q)
+                if r == 1
+                else self.np.concatenate([state, self.np.zeros_like(state)], q)
+            )
         return state
 
     def collapse_state(self, state, qubits, shot, nqubits, normalize=True):

diff --git a/tests/test_backends_clifford.py b/tests/test_backends_clifford.py
@@ -5,7 +5,7 @@
 import numpy as np
 import pytest
 
-from qibo import Circuit, gates, get_backend, set_backend
+from qibo import Circuit, gates, set_backend
 from qibo.backends import (
     CliffordBackend,
     GlobalBackend,