Update src/orquestra/opt/optimizers/tensor_train_optimizer.py

Co-authored-by: Athena Caesura <athena.caesura@zapatacomputing.com>
zapata-engineering · May 1, 2023 · 2796ee9 · 2796ee9
1 parent 8796e9c
commit 2796ee9
Showing 1 changed file with 12 additions and 12 deletions.
diff --git a/src/orquestra/opt/optimizers/tensor_train_optimizer.py b/src/orquestra/opt/optimizers/tensor_train_optimizer.py
@@ -124,18 +124,18 @@ def __call__(self, parameters: np.ndarray) -> float:
                         self.candidates[i]["bounds"] = _get_tighter_bounds(
                             parameters, self.bounds
                         )
-                        break
-                    self.candidates.insert(
-                        i,
-                        {
-                            "cost": cost,
-                            "parameters": parameters,
-                            "bounds": _get_tighter_bounds(parameters, self.bounds),
-                        },
-                    )
-                    self.candidates.pop()
-                    self.candidates_hashes.insert(i, parameters_hash)
-                    self.candidates_hashes.pop()
+                    else:
+                        self.candidates.insert(
+                            i,
+                            {
+                                "cost": cost,
+                                "parameters": parameters,
+                                "bounds": _get_tighter_bounds(parameters, self.bounds),
+                            },
+                        )
+                        self.candidates.pop()
+                        self.candidates_hashes.insert(i, parameters_hash)
+                        self.candidates_hashes.pop()
                     break
             return cost
         candidate = self.candidates[self.candidate_index]