Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
File size: 1,283 Bytes
c0ec7e6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
from typing import Literal
from .atompairs import GetAtomPairFPs
from .avalonfp import GetAvalonFPs
from .rdkitfp import GetRDkitFPs
from .morganfp import GetMorganFPs
from .estatefp import GetEstateFPs
from .maccskeys import GetMACCSFPs
from .pharmErGfp import GetPharmacoErGFPs
from .pharmPointfp import GetPharmacoPFPs
from .pubchemfp import GetPubChemFPs
from .torsions import GetTorsionFPs
from .mhfp6 import GetMHFP6
# from .map4 import GetMAP4
from rdkit import Chem
from deepscreen import get_logger
log = get_logger(__name__)
FP_MAP = {
'MorganFP': GetMorganFPs,
'RDkitFP': GetRDkitFPs,
'AtomPairFP': GetAtomPairFPs,
'TorsionFP': GetTorsionFPs,
'AvalonFP': GetAvalonFPs,
'EstateFP': GetEstateFPs,
'MACCSFP': GetMACCSFPs,
'PharmacoErGFP': GetPharmacoErGFPs,
'PharmacoPFP': GetPharmacoPFPs,
'PubChemFP': GetPubChemFPs,
'MHFP6': GetMHFP6,
# 'MAP4': GetMAP4,
}
def smiles_to_fingerprint(smiles, fingerprint: Literal[tuple(FP_MAP.keys())], **kwargs):
func = FP_MAP[fingerprint]
try:
mol = Chem.MolFromSmiles(smiles)
arr = func(mol, **kwargs)
return arr
except Exception as e:
log.warning(f"Failed to convert SMILES ({smiles}) to {fingerprint} due to {str(e)}")
return None
|