fix: args in scipy optimizers

qiboteam · Jun 28, 2024 · e46995c · e46995c
1 parent 2f541d1
commit e46995c
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/src/boostvqe/utils.py b/src/boostvqe/utils.py
@@ -379,38 +379,38 @@ def optimize_D(
                 func=loss_function_D,
                 x0=params,
                 bounds=bounds,
-                args=(gci,),
+                args=(gci, eo_d_type, mode),
                 maxiter=maxiter,
             )
         elif method == "differential_evolution":
             opt_results = optimize.differential_evolution(
                 func=loss_function_D,
                 x0=params,
                 bounds=bounds,
-                args=(gci,),
+                args=(gci, eo_d_type, mode),
                 maxiter=maxiter,
             )
         elif method == "DIRECT":
             opt_results = optimize.direct(
                 func=loss_function_D,
                 bounds=bounds,
-                args=(gci,),
+                args=(gci, eo_d_type, mode),
                 maxiter=maxiter,
             )
         elif method == "basinhopping":
             opt_results = optimize.basinhopping(
                 func=loss_function_D,
                 x0=params,
                 niter=maxiter,
-                minimizer_kwargs={"method": "Powell", "args": (gci,)},
+                minimizer_kwargs={"method": "Powell", "args": (gci, eo_d_type, mode)},
             )
         # scipy local minimizers
         else:
             opt_results = optimize.minimize(
                 fun=loss_function_D,
                 x0=params,
                 bounds=bounds,
-                args=(gci,),
+                args=(gci, eo_d_type, mode),
                 method=method,
                 options={"disp": 1, "maxiter": maxiter},
             )