From abb3c9b16e50641186e7ae1ee9ef84b364afab5f Mon Sep 17 00:00:00 2001 From: "Jan C. Brammer" Date: Fri, 29 Sep 2023 16:47:40 +0000 Subject: [PATCH] Retrieve number of partitions more intuitively --- tucan/canonicalization.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tucan/canonicalization.py b/tucan/canonicalization.py index 0aa0f89..cfa13e5 100644 --- 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]: