fixed kmeans bug stopping tests working

Author: chrisholder

aeon/clustering/_k_means.py

@@ -268,7 +268,7 @@ def _fit_one_init(self, X: np.ndarray) -> tuple:
             prev_inertia = curr_inertia
             prev_labels = curr_labels
 
-            if change_in_centres < self.tol:
+            if change_in_centres < self.tol or (i + 1) == self.max_iter:
                 break
 
             # Compute new cluster centres

aeon/testing/estimator_checking/_yield_clustering_checks.py

@@ -108,12 +108,12 @@ def check_clusterer_output(estimator, datatype):
     estimator = _clone_estimator(estimator)
 
     # run fit and predict
-    estimator.fit(
-        FULL_TEST_DATA_DICT[datatype]["train"][0],
-        FULL_TEST_DATA_DICT[datatype]["train"][1],
-    )
+    data = FULL_TEST_DATA_DICT[datatype]["train"][0]
+    estimator.fit(data)
     assert hasattr(estimator, "labels_")
     assert isinstance(estimator.labels_, np.ndarray)
+    assert np.array_equal(estimator.labels_, estimator.fit_predict(data))
+    assert np.array_equal(estimator.labels_, estimator.predict(data))
 
     y_pred = estimator.predict(FULL_TEST_DATA_DICT[datatype]["test"][0])