Merge remote-tracking branch 'origin/r_cluster_branch' into r_cluster…

…_branch
aeon-toolkit · Nov 27, 2024 · be5b021 · be5b021
2 parents 8692ca0 + b07777d
commit be5b021
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/aeon/clustering/tests/test_r_cluster.py b/aeon/clustering/tests/test_r_cluster.py
@@ -13,8 +13,9 @@
 
 expected_results = [0, 0, 0, 0, 0]
 
+
 @pytest.mark.skipif(
-    not _check_estimator_deps( RCluster, severity="none"),
+    not _check_estimator_deps(RCluster, severity="none"),
     reason="skip test if required soft dependencies not available",
 )
 def test_kernel_k_means():
@@ -24,11 +25,10 @@ def test_kernel_k_means():
     X_train, y_train = load_basic_motions(split="train")
     X_test, y_test = load_basic_motions(split="test")
 
-    r_cluster =  RCluster( n_clusters=2)
+    r_cluster = RCluster(n_clusters=2)
     r_cluster.fit(X_train[0:max_train])
-    test_shape_result =  r_cluster.predict(X_test[0:max_train])
-
+    test_shape_result = r_cluster.predict(X_test[0:max_train])
 
     assert np.array_equal(test_shape_result, expected_results)
     assert r_cluster.n_iter_ == expected_iters
-    assert np.array_equal( r_cluster.labels_, expected_labels)
+    assert np.array_equal(r_cluster.labels_, expected_labels)