diff --git a/aeon/classification/distance_based/_time_series_neighbors.py b/aeon/classification/distance_based/_time_series_neighbors.py index 57970c24cb..b8ad86a7f6 100644 --- a/aeon/classification/distance_based/_time_series_neighbors.py +++ b/aeon/classification/distance_based/_time_series_neighbors.py @@ -8,6 +8,8 @@ __maintainer__ = [] __all__ = ["KNeighborsTimeSeriesClassifier"] +from typing import List + import numpy as np from aeon.classification.base import BaseClassifier @@ -69,12 +71,12 @@ class KNeighborsTimeSeriesClassifier(BaseClassifier): def __init__( self, - distance="dtw", - distance_params=None, - n_neighbors=1, - weights="uniform", - n_jobs=1, - ): + distance: str = "dtw", + distance_params: dict = None, + n_neighbors: int = 1, + weights: str = "uniform", + n_jobs: int = 1, + ) -> None: self.distance = distance self.distance_params = distance_params self.n_neighbors = n_neighbors @@ -214,7 +216,7 @@ def _kneighbors(self, X): return closest_idx, ws @classmethod - def get_test_params(cls, parameter_set="default"): + def get_test_params(cls, parameter_set: str = "default") -> List[dict]: """Return testing parameter settings for the estimator. Parameters