Skip to content

Commit e8b10e6

Browse files
authored
Merge pull request #3 from JerryChen97/deprecate-device-shots
deprecate device shots
2 parents c64a301 + 34f706e commit e8b10e6

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/iqpopt/iqp_optimizer.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -178,8 +178,9 @@ def sample(self, params: jnp.ndarray, init_coefs: list = None, shots: int = 1) -
178178
return jnp.array(samples)
179179

180180
else:
181-
dev = qml.device(self.device, wires=self.n_qubits, shots=shots)
181+
dev = qml.device(self.device, wires=self.n_qubits)
182182

183+
@qml.set_shots(shots)
183184
@qml.qnode(dev)
184185
def sample_circuit(params):
185186
self.iqp_circuit(params, init_coefs)

0 commit comments

Comments
 (0)