graph now uses string keys instead of ints

Author: sfluegel05

chebifier/prediction_models/c3p_predictor.py

@@ -39,7 +39,7 @@ def predict_smiles_tuple(self, smiles_list: tuple[str]) -> list:
                 chebi_id
             ] = result.is_match
             if result.is_match and self.chebi_graph is not None:
-                for parent in list(self.chebi_graph.predecessors(int(chebi_id))):
+                for parent in list(self.chebi_graph.predecessors(chebi_id)):
                     result_reformatted[smiles_list.index(result.input_smiles)][
                         str(parent)
                     ] = 1

chebifier/prediction_models/chemlog_predictor.py

@@ -63,7 +63,7 @@ def predict_smiles_tuple(self, smiles_list: tuple[str]) -> list:
                 sample_additions = dict()
                 for cls in sample:
                     if sample[cls] == 1:
-                        successors = list(self.chebi_graph.predecessors(int(cls)))
+                        successors = list(self.chebi_graph.predecessors(cls))
                         if successors:
                             for succ in successors:
                                 sample_additions[str(succ)] = 1
@@ -114,7 +114,7 @@ def predict_smiles(self, smiles: str) -> Optional[dict]:
             indirect_pos_labels = [
                 str(pr)
                 for label in pos_labels
-                for pr in self.chebi_graph.predecessors(int(label))
+                for pr in self.chebi_graph.predecessors(label)
             ]
             pos_labels = list(set(pos_labels + indirect_pos_labels))
         return {