Fixes for failing tests in PR

aeon-toolkit · Apr 25, 2024 · 07b46c5 · 07b46c5
1 parent d663a3d
commit 07b46c5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/aeon/classification/distance_based/_time_series_neighbors.py b/aeon/classification/distance_based/_time_series_neighbors.py
@@ -12,6 +12,7 @@
 
 from aeon.classification.base import BaseClassifier
 from aeon.distances import get_distance_function
+from typing import List
 
 WEIGHTS_SUPPORTED = ["uniform", "distance"]
 
@@ -214,7 +215,7 @@ def _kneighbors(self, X):
         return closest_idx, ws
 
     @classmethod
-    def get_test_params(cls, parameter_set: str = "default") -> list[dict]:
+    def get_test_params(cls, parameter_set: str = "default") -> List[dict]:
         """Return testing parameter settings for the estimator.
 
         Parameters