Get the gene keys and gene list keys from the token dictionary instead of medians (#304)

- Get the gene keys and gene list keys from the token dictionary instead of medians (faab0d7678ab0347fae033a620c388ca970b59e0)


Co-authored-by: Han Chen <[email protected]>

geneformer/tokenizer.py

@@ -133,7 +133,7 @@ class TranscriptomeTokenizer:
             self.gene_token_dict = pickle.load(f)
 
         # gene keys for full vocabulary
-        self.gene_keys = list(self.gene_median_dict.keys())
+        self.gene_keys = list(self.gene_token_dict.keys())
 
         # protein-coding and miRNA gene list dictionary for selecting .loom rows for tokenization
         self.genelist_dict = dict(zip(self.gene_keys, [True] * len(self.gene_keys)))