Update trainer.py

trainer.py

@@ -733,7 +733,10 @@ class Trainer(object):
             self.G2.load_state_dict(torch.load(G2_path, map_location=lambda storage, loc: storage))           
         
         
-        drug_smiles = [line for line in open("data/akt_test.smi", 'r').read().splitlines()]
+        if self.submodel == "NoTarget":
+            drug_smiles = [line for line in open("DrugGEN/data/chembl_test.smi", 'r').read().splitlines()]
+        else:
+            drug_smiles = [line for line in open("DrugGEN/data/akt_test.smi", 'r').read().splitlines()]
         
         drug_mols = [Chem.MolFromSmiles(smi) for smi in drug_smiles]
         drug_scaf = [MurckoScaffold.GetScaffoldForMol(x) for x in drug_mols]