ctheodoris commited on
Commit
0efc3b3
·
verified ·
1 Parent(s): df1ed01

ensure anndata/zarr returns data rather than dir if not collapse_gene_ids

Browse files
Files changed (1) hide show
  1. geneformer/tokenizer.py +1 -1
geneformer/tokenizer.py CHANGED
@@ -240,7 +240,7 @@ def sum_ensembl_ids(
240
  gene for gene in ensembl_ids if gene in gene_token_dict.keys()
241
  ]
242
  if len(ensembl_id_check) == len(set(ensembl_id_check)):
243
- return data_directory
244
  else:
245
  raise ValueError("Error: data Ensembl IDs non-unique.")
246
 
 
240
  gene for gene in ensembl_ids if gene in gene_token_dict.keys()
241
  ]
242
  if len(ensembl_id_check) == len(set(ensembl_id_check)):
243
+ return data
244
  else:
245
  raise ValueError("Error: data Ensembl IDs non-unique.")
246