Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion aeon/classification/sklearn/_continuous_interval_tree.py
Original file line number Diff line number Diff line change
Expand Up @@ -483,7 +483,7 @@ def tree_node_splits_and_gain(self) -> tuple[list[int], list[float]]:

return splits, gains

def _find_splits_gain(self, node: type[_TreeNode], splits: list, gains: list):
def _find_splits_gain(self, node: _TreeNode, splits: list, gains: list):
"""Recursively find the split and information gain for each tree node."""
splits.append(node.best_split)
gains.append(node.best_gain)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -453,7 +453,7 @@ def _fit_estimator(

return tree, pcas, groups, X_t if save_transformed_data else None

def _predict_proba_for_estimator(self, X, clf: int, pcas: type[PCA], groups):
def _predict_proba_for_estimator(self, X, clf: int, pcas: PCA, groups):
X_t = np.concatenate(
[pcas[i].transform(X[:, group]) for i, group in enumerate(groups)], axis=1
)
Expand Down
2 changes: 1 addition & 1 deletion aeon/regression/sklearn/_rotation_forest_regressor.py
Original file line number Diff line number Diff line change
Expand Up @@ -352,7 +352,7 @@ def _fit_estimator(

return tree, pcas, groups, X_t if save_transformed_data else None

def _predict_for_estimator(self, X, clf: int, pcas: type[PCA], groups):
def _predict_for_estimator(self, X, clf: int, pcas: PCA, groups):
X_t = np.concatenate(
[pcas[i].transform(X[:, group]) for i, group in enumerate(groups)], axis=1
)
Expand Down