fix sort

Author: juanitorduz

pymc_bart/bart.py

@@ -15,6 +15,7 @@
 #   limitations under the License.
 
 from multiprocessing import Manager
+from typing import List, Optional, Tuple
 
 import numpy as np
 import numpy.typing as npt
@@ -23,11 +24,8 @@
 from pymc.distributions.distribution import Distribution, _moment
 from pymc.logprob.abstract import _logprob
 from pytensor.tensor.random.op import RandomVariable
-from typing import List, Optional, Tuple
-
-
-from .utils import _sample_posterior, TensorLike
 
+from .utils import TensorLike, _sample_posterior
 
 __all__ = ["BART"]
 

pymc_bart/utils.py

@@ -1,6 +1,7 @@
 """Utility function for variable selection and bart interpretability."""
 
 import warnings
+from typing import List, Optional, Tuple, Union
 
 import arviz as az
 import matplotlib.pyplot as plt
@@ -11,9 +12,8 @@
 from scipy.interpolate import griddata
 from scipy.signal import savgol_filter
 from scipy.stats import norm, pearsonr
-from typing import List, Optional, Tuple, Union
-from .tree import Tree
 
+from .tree import Tree
 
 TensorLike = Union[npt.NDArray[np.float_], pt.TensorVariable]