fix if statement on state

pasqal-io · Nov 11, 2024 · fb65e0d · fb65e0d
1 parent 86385f8
commit fb65e0d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/qadence/measurements/shadow.py b/qadence/measurements/shadow.py
@@ -125,7 +125,7 @@ def classical_shadow(
     unitary_ids = np.random.randint(0, 3, size=(shadow_size, circuit.n_qubits))
     all_unitaries = extract_unitaries(unitary_ids, circuit.n_qubits)
     shadow: list = list()
-    batchsize = state.shape[0] if state else 1
+    batchsize = state.shape[0] if state is not None else 1
     for i in range(shadow_size):
         random_unitary_block = all_unitaries[i]
         rotated_circuit = rotate(circuit, (random_unitary_block, 1.0))