Spaces:
Runtime error
Runtime error
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)
|