diff --git a/skglm/estimators.py b/skglm/estimators.py index 85a735ff..ed1f68c8 100644 --- a/skglm/estimators.py +++ b/skglm/estimators.py @@ -21,7 +21,7 @@ from skglm.utils.jit_compilation import compiled_clone from skglm.solvers import AndersonCD, MultiTaskBCD, GroupBCD from skglm.datafits import (Cox, Quadratic, Logistic, QuadraticSVC, - QuadraticMultiTask, QuadraticGroup, QuadraticHessian) + QuadraticMultiTask, QuadraticGroup,) from skglm.penalties import (L1, WeightedL1, L1_plus_L2, L2, WeightedGroupL2, MCPenalty, WeightedMCPenalty, IndicatorBox, L2_1) from skglm.utils.data import grp_converter diff --git a/skglm/tests/test_datafits.py b/skglm/tests/test_datafits.py index e5ca4323..cdd77df4 100644 --- a/skglm/tests/test_datafits.py +++ b/skglm/tests/test_datafits.py @@ -237,5 +237,6 @@ def test_HessianQuadratic(): # check that it's not just because we got alpha too high and thus 0 coef np.testing.assert_array_less(0.1, np.max(np.abs(qpl1.coef_))) + if __name__ == '__main__': pass