Minor bug fixed

kalininalab · May 21, 2024 · 36e529d · 36e529d
1 parent 3bd0d98
commit 36e529d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_clustering.py b/tests/test_clustering.py
@@ -121,7 +121,7 @@ def test_force_clustering():
         [0.1, 0.2, 0.4, 0.6, 1]
     ])
     dataset.classes = {0: 0}
-    dataset.cluster_stratification = {"cluster1": np.array([0]), "cluster2": np.array([0]), "cluster3": np.array([0]), "cluster4": np.array([0]), "5": np.array([0])}
+    dataset.cluster_stratification = {"cluster1": np.array([0]), "cluster2": np.array([0]), "cluster3": np.array([0]), "cluster4": np.array([0]), "cluster5": np.array([0])}
     dataset.num_clusters = 3
 
     # Call the force_clustering function