add scvi test dataset

mims-harvard · Feb 6, 2025 · e70285e · e70285e
1 parent f7a8e9c
commit e70285e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tdc/metadata.py b/tdc/metadata.py
@@ -958,6 +958,7 @@ def get_task2category():
     "evebio_pharmone_v1_target_table": "tab",
     "cellxgene_sample_small": "h5ad",
     "scgpt_vocab": "json",
+    "scvi_test_dataset": "h5ad",
 }
 
 name2id = {
@@ -1166,6 +1167,7 @@ def get_task2category():
     "evebio_pharmone_v1_target_table": 10741537,
     "cellxgene_sample_small": 10806522,
     "scgpt_vocab": 10809431,
+    "scvi_test_dataset": 10872157,
 }
 
 oracle2type = {