pub_soft/PocketFlow
3
0
Fork 0
mirror of https://github.com/Saoge123/PocketFlow.git synced 2026-06-04 12:44:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update generate.py

Browse Source
This commit is contained in:
Saoge123
2023-12-04 09:14:13 +08:00
committed by GitHub
parent b70b9979c2
commit b983006478
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pocket_flow/generate.py
View File

@@ -357,14 +357,12 @@ class Generate(object):
data_clone = data.clone().detach()
out = self.run(data_clone)
if out:
mol, mol_NoModify = out
mol, _ = out
del data_clone
if mol is not None:
#print(len(mol.GetAtoms()))
mol.SetProp('_Name', 'No_%s-%s'%(valid_conuter, out_dir))
mol_NoModify.SetProp('_Name', 'No_%s-%s' %(valid_conuter, out_dir))
smi = Chem.MolToSmiles(mol)
smi_NoModify = Chem.MolToSmiles(mol_NoModify)
if with_print:
print(smi)
with open(out_dir+'generated.sdf', 'a') as sdf_writer: