Update api.py

Author: cthoyt

src/semra/api.py

@@ -566,7 +566,9 @@ def prioritize(
     }
     # Gather all the references by CURIE
     curie_to_reference: dict[str, Reference] = {
-        reference.curie: reference for mapping in mappings for reference in (mapping.subject, mapping.object)
+        reference.curie: reference
+        for mapping in mappings
+        for reference in (mapping.subject, mapping.object)
     }
 
     exact_mappings = len(mappings)
@@ -581,8 +583,7 @@ def prioritize(
         o = get_priority_reference(component_references, priority)
         if o is None:
             continue
-        for s_curie in component:
-            s = curie_to_reference[s_curie]
+        for s in component_references:
             if s == o:  # don't add self-edges
                 continue
             if not graph.has_edge(s, o):
@@ -594,7 +595,7 @@ def prioritize(
                     "that in a given component, it is a full clique (i.e., there are edges "
                     "in both directions between all nodes)"
                 )
-            rv.append(mappings_by_subj_obj[s_curie, o.curie])
+            rv.append(mappings_by_subj_obj[s.curie, o.curie])
 
     # sort such that the mappings are ordered by object by priority order
     # then identifier of object, then subject prefix in alphabetical order