fix typo

src/dataset.py

@@ -39,7 +39,7 @@ class DrugRetrieval(Dataset):
         self.remove_batch = True
 
         assert os.path.exists(os.path.join(self.data_path, f'processed/{drug_encoder}_encoding.pickle')), 'Drug embeddings not available.'
-        assert os.path.exists(f'data/Lenselink/processed/{target_encoder}_encoding_train.pickle')), 'Context target embeddings not available.'
+        assert os.path.exists(f'data/Lenselink/processed/{target_encoder}_encoding_train.pickle'), 'Context target embeddings not available.'
 
         # Drugs
         emb_dict = self.get_drug_embeddings(encoder_name=drug_encoder)
@@ -77,7 +77,7 @@ class DrugRetrieval(Dataset):
         return embeddings
 
     def get_target_embeddings(self, encoder_name):
-        with open(f'data/Lenselink/processed/{encoder_name}_encoding_train.pickle'), 'rb') as handle:
+        with open(f'data/Lenselink/processed/{encoder_name}_encoding_train.pickle', 'rb') as handle:
             embeddings = pickle.load(handle)
         return embeddings