Retrieve number of partitions more intuitively

TUCAN-nest · Aug 23, 2024 · abb3c9b · abb3c9b
1 parent 02aa89c
commit abb3c9b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tucan/canonicalization.py b/tucan/canonicalization.py
@@ -23,7 +23,7 @@ def partition_molecule_by_attribute(m: nx.Graph, attribute: str) -> nx.Graph:
 
 
 def get_number_of_partitions(m: nx.Graph) -> int:
-    return max(nx.get_node_attributes(m, PARTITION).values())
+    return len(set(nx.get_node_attributes(m, PARTITION).values()))
 
 
 def refine_partitions(m: nx.Graph) -> Generator[nx.Graph, None, None]: