Christina Theodoris commited on
Commit
86fe0dd
1 Parent(s): 94095d1

refer to token dictionary in self

Browse files
Files changed (1) hide show
  1. 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()}