fix test

aeon-toolkit · Feb 28, 2025 · 4c95abb · 4c95abb
1 parent 5f74e4e
commit 4c95abb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aeon/classification/distance_based/tests/test_time_series_neighbors.py b/aeon/classification/distance_based/tests/test_time_series_neighbors.py
@@ -45,7 +45,7 @@ def test_knn_on_unit_test(distance_key):
     # load arrowhead data for unit tests
     X_train, y_train = load_unit_test(split="train")
     X_test, y_test = load_unit_test(split="test")
-    knn = KNeighborsTimeSeriesClassifier(distance=distance_key, n_neighbors=3)
+    knn = KNeighborsTimeSeriesClassifier(distance=distance_key)
     knn.fit(X_train, y_train)
     pred = knn.predict(X_test)
     correct = 0