From 5a5f587b4c04265af213f742f12d8e48c4071e51 Mon Sep 17 00:00:00 2001 From: Max Wang Date: Thu, 24 Aug 2023 10:57:28 -0400 Subject: [PATCH] Run black --- strider/trapi.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/strider/trapi.py b/strider/trapi.py index 069ba195..a23e6900 100644 --- a/strider/trapi.py +++ b/strider/trapi.py @@ -151,7 +151,12 @@ def fix_kedge(kedge: Edge, curie_map: dict[str, str]): kedge.object = curie_map.get(kedge.object, [kedge.object])[0] -def fix_results(message: Message, curie_map: dict[str, str], kp_id: str, logger: logging.Logger): +def fix_results( + message: Message, + curie_map: dict[str, str], + kp_id: str, + logger: logging.Logger, +): """Replace curie with preferred, if possible.""" for result in message.results or []: for qnode_id, node_bindings in result.node_bindings.items(): @@ -161,7 +166,9 @@ def fix_results(message: Message, curie_map: dict[str, str], kp_id: str, logger: if node_binding.query_id is None: continue # set node binding query id to preferred prefix - query_id = curie_map.get(node_binding.query_id, [node_binding.query_id])[0] + query_id = curie_map.get( + node_binding.query_id, [node_binding.query_id] + )[0] if node_binding.id == query_id: # remove query_id if equivalent to id logger.info(