From 35a40d0818f1f67cfb21725f520eb435eb6e1423 Mon Sep 17 00:00:00 2001
From: Franck Charras <29153872+fcharras@users.noreply.github.com>
Date: Wed, 17 Jan 2024 11:55:13 +0100
Subject: [PATCH] Fix lsqr

---
 benchmarks/ridge/objective.py | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/benchmarks/ridge/objective.py b/benchmarks/ridge/objective.py
index 0c34844..9f7043e 100644
--- a/benchmarks/ridge/objective.py
+++ b/benchmarks/ridge/objective.py
@@ -93,10 +93,7 @@ def evaluate_result(self, weights, intercept, n_iter, **solver_parameters):
         )
 
         if not (isinstance(n_iter, numbers.Number) or (n_iter is None)):
-            n_iter = set(n_iter)
-            if len(n_iter) > 1:
-                n_iter = max(n_iter)
-            n_iter = n_iter.pop()
+            n_iter = max(n_iter)
 
         # NB: str for n_iter is a more practical type because it enables
         # using missing values for solvers for which it doesn't apply