From cb1965d2d8f46ff18304539da6699373fa052b5e Mon Sep 17 00:00:00 2001
From: Miles Smith <miles-smith@omrf.org>
Date: Tue, 12 Jul 2022 17:40:06 +0000
Subject: [PATCH] change available human atlases to match those in json

---
 scnym/api.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/scnym/api.py b/scnym/api.py
index d39dc54..bdfca1a 100644
--- a/scnym/api.py
+++ b/scnym/api.py
@@ -54,7 +54,8 @@
 REFERENCE_JSON = "https://storage.googleapis.com/calico-website-scnym-storage/link_tables/cell_atlas.json"
 
 ATLAS_ANNOT_KEYS = {
-    "human": "celltype",
+    "human_gtex": "cell_type",
+    "human_gtex_immune": "cell_type",
     "mouse": "cell_ontology_class",
     "rat": "cell_ontology_class",
 }
@@ -822,6 +823,11 @@ def atlas2target(
         names for the relevant species to match the atlas.
         e.g. `"Gapdh`" for mouse or `"GAPDH"` for human, rather
         than Ensembl gene IDs or another gene annotation.
+    species : str
+        One of "human_gtex", "human_gtex_immune", "mouse",
+        or "rat".
+    key_added: str
+        Name to use for the column where cell labels are added
 
     Returns
     -------