Another try at fixing this

Author: Old-Shatterhand

datasail/solver/overflow.py

@@ -52,7 +52,7 @@ def check_dataset(
 
 
 def check_points(dataset, split_ratios, split_names, i: int):
-    sorted_points = sorted(dataset.weights.items(), key=lambda x: x[1], reverse=True)
+    sorted_points = sorted([dataset.weights[name] for name in dataset.names], key=lambda x: x[1], reverse=True)
     total_weight = sum(x[1] for x in sorted_points[i:])
     if [x[1] / total_weight for x in sorted_points[i:len(split_ratios)]] <= sorted(split_ratios, reverse=True):
         return None
@@ -64,7 +64,7 @@ def check_points(dataset, split_ratios, split_names, i: int):
 
 
 def check_clusters(dataset, split_ratios, split_names, strategy: Literal["break", "assign"], linkage: Literal["average", "single", "complete"], i: int):
-    sorted_clusters = sorted(dataset.cluster_weights.items(), key=lambda x: x[1], reverse=True)
+    sorted_clusters = sorted([dataset.cluster_weights[name] for name in dataset.cluster_names], key=lambda x: x[1], reverse=True)
     total_weight = sum(x[1] for x in sorted_clusters[i:])
     if [x[1] / total_weight for x in sorted_clusters[i:len(split_ratios)]] <= sorted(split_ratios, reverse=True):
         return None