diff --git a/pytest/test_args.py b/pytest/test_args.py index e460fd8..c5c151a 100644 --- a/pytest/test_args.py +++ b/pytest/test_args.py @@ -32,7 +32,9 @@ def test_numeric_solver(model, solver_creator): from skscope.numeric_solver import convex_solver_BFGS - solver = solver_creator(model["n_features"], model["n_informative"], numeric_solver=convex_solver_BFGS) + solver = solver_creator( + model["n_features"], model["n_informative"], numeric_solver=convex_solver_BFGS + ) solver.solve(model["loss"], jit=True) assert set(model["support_set"]) == set(solver.get_support()) @@ -167,7 +169,9 @@ def test_scope_hessian(): def test_scope_dynamic_max_exchange_num(): - solver = ScopeSolver(linear["n_features"], linear["n_informative"], is_dynamic_max_exchange_num=False) + solver = ScopeSolver( + linear["n_features"], linear["n_informative"], is_dynamic_max_exchange_num=False + ) solver.solve(linear["loss"], jit=True) assert set(linear["support_set"]) == set(solver.support_set)