Fix cupy

qiboteam · Jan 22, 2023 · cfba10c · cfba10c
1 parent 80139d4
commit cfba10c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/qibo/gates/channels.py b/src/qibo/gates/channels.py
@@ -303,9 +303,9 @@ def __init__(self, q, lam=0):
 
     def apply_density_matrix(self, backend, state, nqubits):
         lam = self.init_kwargs["lam"]
-        return (1 - lam) * backend.cast(state) + lam / 2**nqubits * I(
-            *range(nqubits)
-        ).asmatrix(backend)
+        return (1 - lam) * backend.cast(state) + lam / 2**nqubits * backend.cast(
+            I(*range(nqubits)).asmatrix(backend)
+        )
 
     def apply(self, backend, state, nqubits):
         num_qubits = len(self.target_qubits)