sort imports

src/regmod/_typing.py

@@ -1,6 +1,7 @@
-from typing import Any, Callable
 from collections.abc import Iterable
-from numpy.typing import ArrayLike, NDArray
-from pandas import DataFrame
+from typing import Any, Callable
+
 from jax import Array as JaxArray
 from msca.linalg.matrix import Matrix
+from numpy.typing import ArrayLike, NDArray
+from pandas import DataFrame

src/regmod/function.py

@@ -3,10 +3,11 @@
 """
 
 from dataclasses import dataclass, field
-from regmod._typing import Callable
 
 import numpy as np
 
+from regmod._typing import Callable
+
 
 @dataclass
 class SmoothFunction:

src/regmod/models/binomial.py

@@ -5,8 +5,8 @@
 import numpy as np
 from scipy.stats import binom
 
+from regmod._typing import Callable, DataFrame, NDArray
 from regmod.optimizer import msca_optimize
-from regmod._typing import Callable, NDArray, DataFrame
 
 from .model import Model
 from .utils import model_post_init

src/regmod/models/gaussian.py

@@ -5,8 +5,8 @@
 import numpy as np
 from scipy.stats import norm
 
+from regmod._typing import Callable, DataFrame, NDArray
 from regmod.optimizer import msca_optimize
-from regmod._typing import Callable, NDArray, DataFrame
 
 from .model import Model
 from .utils import model_post_init

src/regmod/models/model.py

@@ -6,10 +6,10 @@
 from scipy.linalg import block_diag
 from scipy.sparse import csc_matrix
 
+from regmod._typing import Callable, DataFrame, Matrix, NDArray
 from regmod.optimizer import scipy_optimize
 from regmod.parameter import Parameter
 from regmod.utils import sizes_to_slices
-from regmod._typing import Callable, NDArray, DataFrame, Matrix
 
 
 class Model:

src/regmod/models/negativebinomial.py

@@ -7,7 +7,8 @@
 from scipy.special import digamma, loggamma, polygamma
 from scipy.stats import nbinom
 
-from regmod._typing import NDArray, DataFrame
+from regmod._typing import DataFrame, NDArray
+
 from .model import Model
 
 

src/regmod/models/pogit.py

@@ -5,7 +5,8 @@
 import numpy as np
 from scipy.stats import poisson
 
-from regmod._typing import NDArray, DataFrame
+from regmod._typing import DataFrame, NDArray
+
 from .model import Model
 
 

src/regmod/models/poisson.py

@@ -5,8 +5,9 @@
 import numpy as np
 from scipy.stats import poisson
 
+from regmod._typing import Callable, DataFrame, NDArray
 from regmod.optimizer import msca_optimize
-from regmod._typing import Callable, NDArray, DataFrame
+
 from .model import Model
 from .utils import model_post_init
 

src/regmod/models/tobit.py

@@ -7,8 +7,8 @@
 from jax import grad, hessian, jit, lax
 from jax.scipy.stats.norm import logcdf, logpdf
 
-from regmod.function import SmoothFunction
 from regmod._typing import ArrayLike, DataFrame, JaxArray
+from regmod.function import SmoothFunction
 
 from .model import Model
 

src/regmod/models/utils.py

@@ -1,7 +1,8 @@
 import numpy as np
 from msca.linalg.matrix import asmatrix
 from scipy.sparse import csc_matrix
-from regmod._typing import NDArray, Matrix
+
+from regmod._typing import Matrix, NDArray
 
 
 def model_post_init(

src/regmod/models/weibull.py

@@ -5,7 +5,8 @@
 import numpy as np
 from scipy.stats import weibull_min
 
-from regmod._typing import NDArray, DataFrame
+from regmod._typing import DataFrame, NDArray
+
 from .model import Model
 
 

src/regmod/optimizer.py

@@ -6,6 +6,7 @@
 from msca.optim.prox import proj_capped_simplex
 from msca.optim.solver import IPSolver, NTSolver
 from scipy.optimize import LinearConstraint, minimize
+
 from regmod._typing import Callable, NDArray
 
 

src/regmod/parameter.py

@@ -7,10 +7,10 @@
 import numpy as np
 from scipy.linalg import block_diag
 
+from regmod._typing import DataFrame, NDArray
 from regmod.function import SmoothFunction, fun_dict
 from regmod.prior import LinearGaussianPrior, LinearUniformPrior
 from regmod.variable import SplineVariable, Variable
-from regmod._typing import NDArray, DataFrame
 
 
 @dataclass

src/regmod/variable.py

@@ -8,6 +8,7 @@
 import numpy as np
 from xspline import XSpline
 
+from regmod._typing import DataFrame, Iterable, NDArray
 from regmod.prior import (
     GaussianPrior,
     LinearGaussianPrior,
@@ -20,7 +21,6 @@
     UniformPrior,
 )
 from regmod.utils import SplineSpecs
-from regmod._typing import Iterable, NDArray, DataFrame
 
 
 @dataclass