From 1239b9ba0dfa5464a74388943870552d7a7382bb Mon Sep 17 00:00:00 2001 From: TheSalocin Date: Tue, 10 Sep 2024 13:08:25 +0200 Subject: [PATCH] allow returning multiple integer upstream nodes --- datasci_tools/networkx_utils.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/datasci_tools/networkx_utils.py b/datasci_tools/networkx_utils.py index 9ae87cf..1ec9d1f 100644 --- a/datasci_tools/networkx_utils.py +++ b/datasci_tools/networkx_utils.py @@ -1314,7 +1314,10 @@ def upstream_node(G,node,return_single=True): if return_single: raise Exception(f"More than one upstream node for node {node}: {curr_upstream_nodes}") else: - return [k[0] for k in curr_upstream_nodes] + try: + return [k[0] for k in curr_upstream_nodes] + except: + return [k for k in curr_upstream_nodes] else: if not nu.is_array_like(curr_upstream_nodes[0]): return curr_upstream_nodes[0]