fix unexpected behavior of get_terminal_structures with {} structures

BojarLab · Jun 30, 2023 · eaef2bd · eaef2bd
1 parent 656e37d
commit eaef2bd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/build/lib/glycowork/motif/annotate.py b/build/lib/glycowork/motif/annotate.py
@@ -312,7 +312,7 @@ def get_terminal_structures(glycan, libr = None):
     libr = lib
   ggraph = ensure_graph(glycan, libr = libr)
   nodeDict = dict(ggraph.nodes(data = True))
-  return [nodeDict[k]['string_labels']+'('+nodeDict[k+1]['string_labels']+')' for k in ggraph.nodes() if ggraph.degree[k] == 1 and k != max(list(ggraph.nodes()))]
+  return [nodeDict[k]['string_labels']+'('+nodeDict[k+1]['string_labels']+')' for k in list(ggraph.nodes())[:-1] if ggraph.degree[k] == 1 and k+1 in nodeDict.keys()]
 
 
 def create_correlation_network(df, correlation_threshold):

diff --git a/glycowork/motif/annotate.py b/glycowork/motif/annotate.py
@@ -312,7 +312,7 @@ def get_terminal_structures(glycan, libr = None):
     libr = lib
   ggraph = ensure_graph(glycan, libr = libr)
   nodeDict = dict(ggraph.nodes(data = True))
-  return [nodeDict[k]['string_labels']+'('+nodeDict[k+1]['string_labels']+')' for k in ggraph.nodes() if ggraph.degree[k] == 1 and k != max(list(ggraph.nodes()))]
+  return [nodeDict[k]['string_labels']+'('+nodeDict[k+1]['string_labels']+')' for k in list(ggraph.nodes())[:-1] if ggraph.degree[k] == 1 and k+1 in nodeDict.keys()]
 
 
 def create_correlation_network(df, correlation_threshold):