Fix typing

Signed-off-by: Adam Li <[email protected]>
neurodata · Nov 8, 2023 · 1ff7a5c · 1ff7a5c
1 parent 36c5582
commit 1ff7a5c
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/sktree/experimental/mutual_info.py b/sktree/experimental/mutual_info.py
@@ -147,7 +147,7 @@ def mutual_info_ksg(
     algorithm="kd_tree",
     n_jobs: int = -1,
     transform: str = "rank",
-    random_seed: int = None,
+    random_seed: Optional[int] = None,
 ):
     """Compute the generalized (conditional) mutual information KSG estimate.
 

diff --git a/sktree/stats/forestht.py b/sktree/stats/forestht.py
@@ -1,4 +1,4 @@
-from typing import Optional, Callable, Tuple, Union
+from typing import Callable, Optional, Tuple, Union
 
 import numpy as np
 from joblib import Parallel, delayed

diff --git a/sktree/stats/permutationforest.py b/sktree/stats/permutationforest.py
@@ -1,3 +1,5 @@
+from typing import Optional
+
 import numpy as np
 from numpy.typing import ArrayLike
 from sklearn.base import MetaEstimatorMixin, clone, is_classifier
@@ -10,7 +12,6 @@
 from sktree._lib.sklearn.ensemble._forest import BaseForest, ForestClassifier, ForestRegressor
 
 from .utils import METRIC_FUNCTIONS, REGRESSOR_METRICS, _compute_null_distribution_perm
-from typing import Optional
 
 
 class BasePermutationForest(MetaEstimatorMixin):