fixed solver mask

flatironinstitute · Nov 1, 2023 · 61a2511 · 61a2511
1 parent 448d370
commit 61a2511
Showing 1 changed file with 2 additions and 5 deletions.
diff --git 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(