From 5c01e619295db72fb5ddd6760d146086226836e7 Mon Sep 17 00:00:00 2001 From: Caleb Johnson Date: Tue, 11 Apr 2023 19:29:04 -0500 Subject: [PATCH] Fix basis in EFGSS test --- .../test_entanglement_forging_ground_state_solver.py | 6 +++--- 1 file changed, 3 insertions(+), 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 index 3e1e0101d..d99f546ff 100644 --- 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,