correct tsp docstring

qiboteam · Dec 13, 2023 · 40b7526 · 40b7526
1 parent 917df00
commit 40b7526
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/qibo/models/tsp.py b/src/qibo/models/tsp.py
@@ -104,11 +104,13 @@ def qaoa_function_of_layer(layer, distance_matrix):
                 from QAOA
 
                 '''
+                from qibo.oprtimizers.minimizers import ScipyMinimizer
                 small_tsp = TSP(distance_matrix)
                 obj_hamil, mixer = small_tsp.hamiltonians()
                 qaoa = QAOA(obj_hamil, mixer=mixer)
-                best_energy, final_parameters, extra = qaoa.minimize(initial_p=[0.1] * layer,
-                                                     initial_state=initial_state, method='BFGS')
+                opt = ScipyMinimizer({'method': 'BFGS'})
+                best_energy, final_parameters, extra = qaoa.minimize(opt, initial_parameters=[0.1] * layer,
+                                                     initial_state=initial_state)
                 qaoa.set_parameters(final_parameters)
                 quantum_state = qaoa.execute(initial_state)
                 circuit = Circuit(9)

diff --git a/src/qibo/models/variational.py b/src/qibo/models/variational.py
@@ -447,7 +447,7 @@ def minimize(
 
                 from qibo import hamiltonians
                 from qibo.models.utils import cvar, gibbs
-                from qibo.optimizer.minimizers import ScipyMinimizer
+                from qibo.optimizers.minimizers import ScipyMinimizer
 
                 h = hamiltonians.XXZ(3)
                 qaoa = models.QAOA(h)