fix ensembl to seq

BiomedSciAI · Aug 5, 2024 · a16f232 · a16f232
1 parent e904e04
commit a16f232
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gene_benchmark/descriptor.py b/gene_benchmark/descriptor.py
@@ -29,7 +29,7 @@ def _gene_symbol_to_ensemble_ids(
  ids = {}
  for hit in gene_info["hits"]:
  symbol = hit["symbol"]
- ensembl_id = hit.get("ensembl", {}).get("gene", None)
+ ensembl_id = hit["ensembl"]["gene"]
  ids[symbol] = ensembl_id
  return ids