Skip to content

Commit

Permalink
Fix lsqr
Browse files Browse the repository at this point in the history
  • Loading branch information
fcharras committed Jan 17, 2024
1 parent bd78618 commit 35a40d0
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions benchmarks/ridge/objective.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 35a40d0

Please sign in to comment.