From 61a2511d13ea414cbb2ad0a0d4da436e6446652b Mon Sep 17 00:00:00 2001 From: BalzaniEdoardo Date: Wed, 1 Nov 2023 12:38:11 -0400 Subject: [PATCH] fixed solver mask --- src/neurostatslib/solver.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/src/neurostatslib/solver.py b/src/neurostatslib/solver.py index d3f8ac74..78ee12ea 100644 --- a/src/neurostatslib/solver.py +++ b/src/neurostatslib/solver.py @@ -377,7 +377,7 @@ def __init__( solver_name: str, solver_kwargs: Optional[dict] = None, regularizer_strength: float = 1.0, - **kwargs + **kwargs, ): super().__init__(solver_name, solver_kwargs=solver_kwargs) self.regularizer_strength = regularizer_strength @@ -444,10 +444,7 @@ def __init__( solver_kwargs: Optional[dict] = None, regularizer_strength: float = 1.0, ): - super().__init__( - solver_name, - solver_kwargs=solver_kwargs - ) + super().__init__(solver_name, solver_kwargs=solver_kwargs) self.regularizer_strength = regularizer_strength def get_prox_operator(