Fix basis in EFGSS test

Qiskit-Extensions · Apr 12, 2023 · 5c01e61 · 5c01e61
1 parent 5ad45fb
commit 5c01e61
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/entanglement_forging/test_entanglement_forging_ground_state_solver.py b/test/entanglement_forging/test_entanglement_forging_ground_state_solver.py
@@ -44,9 +44,10 @@ def test_entanglement_forging_vqe_hydrogen(self):
             spin=0,
             basis="sto3g",
         )
-        problem = driver.run()
-        problem.basis = ElectronicBasis.AO
+        driver.run()
+        problem = driver.to_problem(basis=ElectronicBasis.AO)
         qcschema = driver.to_qcschema()
+        mo_coeff = get_ao_to_mo_from_qcschema(qcschema).coefficients.alpha["+-"]
 
         # Specify the ansatz and bitstrings
         ansatz = EntanglementForgingAnsatz(
@@ -55,7 +56,6 @@ def test_entanglement_forging_vqe_hydrogen(self):
         )
 
         # Set up the entanglement forging vqe object
-        mo_coeff = get_ao_to_mo_from_qcschema(qcschema).coefficients.alpha["+-"]
         solver = EntanglementForgingGroundStateSolver(
             ansatz=ansatz,
             optimizer=self.optimizer,