From b07777dcb5ef837ed166087a376c7f6512553c0e Mon Sep 17 00:00:00 2001 From: Ramana-Raja Date: Wed, 27 Nov 2024 14:32:19 +0000 Subject: [PATCH] Automatic `pre-commit` fixes --- aeon/clustering/tests/test_r_cluster.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/aeon/clustering/tests/test_r_cluster.py b/aeon/clustering/tests/test_r_cluster.py index 716c2df044..1a8778a1fd 100644 --- 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)