Added checks for different transpiler types

qiskit-community · Dec 12, 2024 · b630a1f · b630a1f
1 parent 94ccb0a
commit b630a1f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/qiskit_machine_learning/neural_networks/estimator_qnn.py b/qiskit_machine_learning/neural_networks/estimator_qnn.py
@@ -182,7 +182,10 @@ def __init__(
                 self.num_virtual_qubits = circuit.num_qubits
             else:
                 circuit = pass_manager.run(circuit)
-                self.num_virtual_qubits = circuit.layout._input_qubit_count
+                if hasattr(circuit.layout, "_input_qubit_count"):
+                    self.num_virtual_qubits = circuit.layout._input_qubit_count
+                else:
+                    self.num_virtual_qubits = circuit.num_qubits
 
         self._org_circuit = circuit
 

diff --git a/qiskit_machine_learning/neural_networks/sampler_qnn.py b/qiskit_machine_learning/neural_networks/sampler_qnn.py
@@ -209,7 +209,10 @@ def __init__(
                 self.num_virtual_qubits = circuit.num_qubits
             else:
                 circuit = pass_manager.run(circuit)
-                self.num_virtual_qubits = circuit.layout._input_qubit_count
+                if hasattr(circuit.layout, "_input_qubit_count"):
+                    self.num_virtual_qubits = circuit.layout._input_qubit_count
+                else:
+                    self.num_virtual_qubits = circuit.num_qubits
 
         self._org_circuit = circuit