Merge pull request #31 from ipqa-research/30-cant-resolve-trans-2-eic…

…osene Better not found on PubChem error
ipqa-research · Nov 24, 2024 · 25060e7 · 25060e7
2 parents f93f355 + cc5f05e
commit 25060e7
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/ugropy/core/get_rdkit_object.py b/ugropy/core/get_rdkit_object.py
@@ -34,9 +34,15 @@ def instantiate_mol_object(
         chem_object = Chem.MolFromSmiles(smiles)
 
     elif identifier_type.lower() == "name":
-        pcp_object = pcp.get_compounds(identifier, identifier_type)[0]
-        smiles = pcp_object.canonical_smiles
-        chem_object = Chem.MolFromSmiles(smiles)
+        try:
+            pcp_object = pcp.get_compounds(identifier, identifier_type)[0]
+            smiles = pcp_object.canonical_smiles
+            chem_object = Chem.MolFromSmiles(smiles)
+        except IndexError:
+            raise ValueError(
+                f"Could not find a molecule with the name '{identifier}' on "
+                "PubChem"
+            )
 
     elif identifier_type.lower() == "mol":
         chem_object = identifier