Skip to content

Commit 76f1718

Browse files
authored
Remove clbits from QuantumNeuralNetwork class (#63)
* remove clbits as sampler primitive cannot take clbits during training * remove classical register from tests
1 parent 9455221 commit 76f1718

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

piqture/neural_networks/quantum_neural_network.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -39,9 +39,10 @@ def __init__(self, num_qubits: int):
3939
raise ValueError("The input num_qubits must be at least 1.")
4040

4141
self.num_qubits = num_qubits
42-
self._circuit = QuantumCircuit(self.num_qubits, self.num_qubits)
42+
self._circuit = QuantumCircuit(self.num_qubits)
4343
self.qr = self._circuit.qubits
44-
self.cr = self._circuit.clbits
44+
# Remove clbits as Sampler cannot take clbits.
45+
# self.cr = self._circuit.clbits
4546

4647
@property
4748
def circuit(self):

tests/neural_networks/test_qcnn.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ def test_value_num_qubits(self, num_qubits):
4444
def test_circuit_property(self, num_qubits):
4545
"""Tests the QCNN circuit."""
4646
qcnn = QCNN(num_qubits)
47-
assert qcnn.circuit == QuantumCircuit(num_qubits, num_qubits)
47+
assert qcnn.circuit == QuantumCircuit(num_qubits)
4848

4949
@pytest.mark.parametrize(
5050
"num_qubits, operations",

0 commit comments

Comments
 (0)