diff --git a/src/sparselm/model/__init__.py b/src/sparselm/model/__init__.py index 922911a..6ea15d8 100644 --- a/src/sparselm/model/__init__.py +++ b/src/sparselm/model/__init__.py @@ -1,4 +1,5 @@ """Classes implementing generalized linear regression Regressors.""" + from ._adaptive_lasso import ( AdaptiveGroupLasso, AdaptiveLasso, diff --git a/src/sparselm/model/_adaptive_lasso.py b/src/sparselm/model/_adaptive_lasso.py index 81c03d2..2c6dada 100644 --- a/src/sparselm/model/_adaptive_lasso.py +++ b/src/sparselm/model/_adaptive_lasso.py @@ -113,8 +113,9 @@ def __init__( max_iter: int = 3, eps: float = 1e-6, tol: float = 1e-10, - update_function: Callable[[NDArray[np.floating], float], ArrayLike] - | None = None, + update_function: ( + Callable[[NDArray[np.floating], float], ArrayLike] | None + ) = None, fit_intercept: bool = False, copy_X: bool = True, warm_start: bool = True, @@ -310,8 +311,9 @@ def __init__( max_iter: int = 3, eps: float = 1e-6, tol: float = 1e-10, - update_function: Callable[[NDArray[np.floating], float], ArrayLike] - | None = None, + update_function: ( + Callable[[NDArray[np.floating], float], ArrayLike] | None + ) = None, standardize: bool = False, fit_intercept: bool = False, copy_X: bool = True, @@ -457,8 +459,9 @@ def __init__( max_iter: int = 3, eps: float = 1e-6, tol: float = 1e-10, - update_function: Callable[[NDArray[np.floating], float], ArrayLike] - | None = None, + update_function: ( + Callable[[NDArray[np.floating], float], ArrayLike] | None + ) = None, standardize: bool = False, fit_intercept: bool = False, copy_X: bool = True, @@ -606,8 +609,9 @@ def __init__( max_iter: int = 3, eps: float = 1e-6, tol: float = 1e-10, - update_function: Callable[[NDArray[np.floating], float], ArrayLike] - | None = None, + update_function: ( + Callable[[NDArray[np.floating], float], ArrayLike] | None + ) = None, standardize: bool = False, fit_intercept: bool = False, copy_X: bool = True, @@ -808,8 +812,9 @@ def __init__( max_iter: int = 3, eps: float = 1e-6, tol: float = 1e-10, - update_function: Callable[[NDArray[np.floating], float], ArrayLike] - | None = None, + update_function: ( + Callable[[NDArray[np.floating], float], ArrayLike] | None + ) = None, standardize: bool = False, fit_intercept: bool = False, copy_X: bool = True, diff --git a/src/sparselm/model/_miqp/__init__.py b/src/sparselm/model/_miqp/__init__.py index 3bc2b8e..c530db9 100644 --- a/src/sparselm/model/_miqp/__init__.py +++ b/src/sparselm/model/_miqp/__init__.py @@ -1,6 +1,5 @@ """MIQP based regression Regressors.""" - from ._best_subset import BestSubsetSelection, RidgedBestSubsetSelection from ._regularized_l0 import L1L0, L2L0, RegularizedL0 diff --git a/tests/test_stepwise.py b/tests/test_stepwise.py index f20a39d..2b1c7c0 100644 --- a/tests/test_stepwise.py +++ b/tests/test_stepwise.py @@ -1,4 +1,5 @@ """Test composite estimator class.""" + import numpy as np import numpy.testing as npt import pytest