Christina Theodoris
commited on
Commit
•
86fe0dd
1
Parent(s):
94095d1
refer to token dictionary in self
Browse files- geneformer/classifier.py +1 -1
geneformer/classifier.py
CHANGED
@@ -233,7 +233,7 @@ class Classifier:
|
|
233 |
# load token dictionary (Ensembl IDs:token)
|
234 |
if self.token_dictionary_file is None:
|
235 |
self.token_dictionary_file = TOKEN_DICTIONARY_FILE
|
236 |
-
with open(token_dictionary_file, "rb") as f:
|
237 |
self.gene_token_dict = pickle.load(f)
|
238 |
|
239 |
self.token_gene_dict = {v: k for k, v in self.gene_token_dict.items()}
|
|
|
233 |
# load token dictionary (Ensembl IDs:token)
|
234 |
if self.token_dictionary_file is None:
|
235 |
self.token_dictionary_file = TOKEN_DICTIONARY_FILE
|
236 |
+
with open(self.token_dictionary_file, "rb") as f:
|
237 |
self.gene_token_dict = pickle.load(f)
|
238 |
|
239 |
self.token_gene_dict = {v: k for k, v in self.gene_token_dict.items()}
|