Skip to content

Commit 88fae29

Browse files
authored
Update utils.py (#45)
Fix file paths so they don't require running from the deep purpose directory
1 parent b336f80 commit 88fae29

1 file changed

Lines changed: 7 additions & 4 deletions

File tree

DeepPurpose/utils.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -20,24 +20,27 @@
2020
from zipfile import ZipFile
2121
import os
2222
import sys
23+
import pathlib
24+
25+
this_dir = str(pathlib.Path(__file__).parent.absolute())
2326

2427
MAX_ATOM = 200
2528
MAX_BOND = 400
2629

2730
# ESPF encoding
28-
vocab_path = './DeepPurpose/ESPF/drug_codes_chembl_freq_1500.txt'
31+
vocab_path = f"{this_dir}/ESPF/drug_codes_chembl_freq_1500.txt"
2932
bpe_codes_drug = codecs.open(vocab_path)
3033
dbpe = BPE(bpe_codes_drug, merges=-1, separator='')
31-
sub_csv = pd.read_csv('./DeepPurpose/ESPF/subword_units_map_chembl_freq_1500.csv')
34+
sub_csv = pd.read_csv(f"{this_dir}/ESPF/subword_units_map_chembl_freq_1500.csv")
3235

3336
idx2word_d = sub_csv['index'].values
3437
words2idx_d = dict(zip(idx2word_d, range(0, len(idx2word_d))))
3538

36-
vocab_path = './DeepPurpose/ESPF/protein_codes_uniprot_2000.txt'
39+
vocab_path = f"{this_dir}/ESPF/protein_codes_uniprot_2000.txt"
3740
bpe_codes_protein = codecs.open(vocab_path)
3841
pbpe = BPE(bpe_codes_protein, merges=-1, separator='')
3942
#sub_csv = pd.read_csv(dataFolder + '/subword_units_map_protein.csv')
40-
sub_csv = pd.read_csv('./DeepPurpose/ESPF/subword_units_map_uniprot_2000.csv')
43+
sub_csv = pd.read_csv(f"{this_dir}/ESPF/subword_units_map_uniprot_2000.csv")
4144

4245
idx2word_p = sub_csv['index'].values
4346
words2idx_p = dict(zip(idx2word_p, range(0, len(idx2word_p))))

0 commit comments

Comments
 (0)