Spaces:
Sleeping
Sleeping
simonduerr
commited on
Commit
·
1be0b1d
1
Parent(s):
ac59745
Update datasets/pdbbind.py
Browse files- datasets/pdbbind.py +2 -0
datasets/pdbbind.py
CHANGED
@@ -332,12 +332,14 @@ class PDBBind(Dataset):
|
|
332 |
print("Reading molecules and generating local structures with RDKit")
|
333 |
for ligand_description in tqdm(self.ligand_descriptions):
|
334 |
mol = MolFromSmiles(ligand_description) # check if it is a smiles or a path
|
|
|
335 |
if mol is not None:
|
336 |
mol = AddHs(mol)
|
337 |
generate_conformer(mol)
|
338 |
ligands_list.append(mol)
|
339 |
else:
|
340 |
mol = read_molecule(ligand_description, remove_hs=False, sanitize=True)
|
|
|
341 |
if not self.keep_local_structures:
|
342 |
mol.RemoveAllConformers()
|
343 |
mol = AddHs(mol)
|
|
|
332 |
print("Reading molecules and generating local structures with RDKit")
|
333 |
for ligand_description in tqdm(self.ligand_descriptions):
|
334 |
mol = MolFromSmiles(ligand_description) # check if it is a smiles or a path
|
335 |
+
print(ligand_description, mol)
|
336 |
if mol is not None:
|
337 |
mol = AddHs(mol)
|
338 |
generate_conformer(mol)
|
339 |
ligands_list.append(mol)
|
340 |
else:
|
341 |
mol = read_molecule(ligand_description, remove_hs=False, sanitize=True)
|
342 |
+
print(mol)
|
343 |
if not self.keep_local_structures:
|
344 |
mol.RemoveAllConformers()
|
345 |
mol = AddHs(mol)
|