Generalize SolutionReadout

lava-nc · Feb 21, 2024 · e2bcb81 · e2bcb81
1 parent 1fcf242
commit e2bcb81
Show file tree

Hide file tree

Showing 4 changed files with 299 additions and 218 deletions.
diff --git a/src/lava/lib/optimization/solvers/qubo/cost_integrator/process.py b/src/lava/lib/optimization/solvers/qubo/cost_integrator/process.py
@@ -23,10 +23,10 @@ class CostIntegrator(AbstractProcess):
         The expected number and topology of the input cost components.
     target_cost: int
         Target cost of the QUBO solver. Once reached, the best_cost,
-        best_timestep, and best_state are spiked out.
+        best_timestep, and best_variable_assignment are spiked out.
     timeout: int
         Timeout of the QUBO solver. Once reached, the best_cost,
-        best_timestep, and best_state are spiked out.
+        best_timestep, and best_variable_assignment are spiked out.
     name : str, optional
         Name of the Process. Default is 'Process_ID', where ID is an
         integer value that is determined automatically.

diff --git a/src/lava/lib/optimization/solvers/qubo/simulated_annealing/process.py b/src/lava/lib/optimization/solvers/qubo/simulated_annealing/process.py
@@ -68,7 +68,7 @@ def __init__(
         self.control_cost_integrator = InPort(shape=shape)
         self.s_sig_out = OutPort(shape=shape)
         self.s_wta_out = OutPort(shape=shape)
-        self.best_state_out = OutPort(shape=shape)
+        self.best_variable_assignment_out = OutPort(shape=shape)
 
         self.spk_hist = Var(
             shape=shape, init=(np.zeros(shape=shape) + init_value).astype(int)