diff --git a/kg_covid_19/transform_utils/string_ppi/string_ppi.py b/kg_covid_19/transform_utils/string_ppi/string_ppi.py index 17dddd49..987d500a 100644 --- a/kg_covid_19/transform_utils/string_ppi/string_ppi.py +++ b/kg_covid_19/transform_utils/string_ppi/string_ppi.py @@ -213,25 +213,25 @@ def run(self, data_file: Optional[str] = None) -> None: protein_node_type, "", "", self.source_name] ) - # if we have an equivalent Uniprot ID for this Ensembl protein - # ID make an xref edge, and a node for the Uniprot ID - if nat_string_id in string_to_uniprot_id_map: - uniprot_curie = \ - f"UniprotKB:{string_to_uniprot_id_map[nat_string_id]}" - write_node_edge_item( - fh=node, - header=self.node_header, - data=[uniprot_curie, "", - protein_node_type, "", "", self.source_name]) - write_node_edge_item( - fh=edge, - header=self.edge_header, - data=[f"ENSEMBL:{protein}", - "biolink:xrefs", - uniprot_curie, - "biolink:xrefs", - "uniprot", - ] + self.extra_header) + # if we have an equivalent Uniprot ID for this Ensembl protein + # ID make an xref edge, and a node for the Uniprot ID + if nat_string_id in string_to_uniprot_id_map: + uniprot_curie = \ + f"UniprotKB:{string_to_uniprot_id_map[nat_string_id]}" + write_node_edge_item( + fh=node, + header=self.node_header, + data=[uniprot_curie, "", + protein_node_type, "", "", self.source_name]) + write_node_edge_item( + fh=edge, + header=self.edge_header, + data=[f"ENSEMBL:{protein}", + "biolink:xrefs", + uniprot_curie, + "biolink:xrefs", + "uniprot", + ] + self.extra_header) # write edge data write_node_edge_item(