Update trainer.py

trainer.py

@@ -734,9 +734,9 @@ class Trainer(object):
         
         
         if self.submodel == "NoTarget":
-            drug_smiles = [line for line in open("data/chembl_test.smi", 'r').read().splitlines()]
+            drug_smiles = [line for line in open("data/chembl_train.smi", 'r').read().splitlines()]
         else:
-            drug_smiles = [line for line in open("data/akt_test.smi", 'r').read().splitlines()]
+            drug_smiles = [line for line in open("data/akt_train.smi", 'r').read().splitlines()]
         
         drug_mols = [Chem.MolFromSmiles(smi) for smi in drug_smiles]
         drug_scaf = [MurckoScaffold.GetScaffoldForMol(x) for x in drug_mols]