From 7f0db0f2083bbebc6ae1c91c755b0ea769880727 Mon Sep 17 00:00:00 2001 From: Marcell Szikszai <64620912+marcellszi@users.noreply.github.com> Date: Wed, 15 May 2024 14:07:04 +0800 Subject: [PATCH] minor fixes to split --- rna3db/split.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/rna3db/split.py b/rna3db/split.py index 6e27ae9..6f990bf 100644 --- a/rna3db/split.py +++ b/rna3db/split.py @@ -61,13 +61,9 @@ def split( output[split_names[-1]]["component_0"] = cluster_json["component_0"] lengths.pop("component_0") - # start with the smallest splits - # splits, split_names = zip(*sorted(zip(splits, split_names), reverse=True)) - - capacities = [int(total_repr_clusters * ratio) for ratio in splits] + capacities = [round(total_repr_clusters * ratio) for ratio in splits] for name, capacity in zip(split_names, capacities): components = find_optimal_components(lengths, capacity) - print(name, capacity, components) for k in sorted(components): lengths.pop(k) output[name][k] = cluster_json[k]