updating the names of the model checkpoints

src/methods/scprint/config.vsh.yaml

@@ -32,18 +32,18 @@ info:
   preferred_normalization: counts
   variants:
     scprint_large:
-      model_name: "large"
+      model_name: "large-v1"
     scprint_medium:
-      model_name: "v2-medium"
+      model_name: "medium-v1.5"
     scprint_small:
-      model_name: "small"
+      model_name: "small-v1"
 
 arguments:
   - name: --model_name
     type: "string"
     description: Which model to use. Not used if --model is provided.
-    choices: ["large", "v2-medium", "small"]
-    default: "v2-medium"
+    choices: ["large-v1", "medium-v.15", "small-v1"]
+    default: "medium5-"
   - name: --model
     type: file
     description: Path to the scPRINT model.

src/methods/scprint/script.py

@@ -13,7 +13,7 @@
 par = {
     "input_train": "resources_test/task_batch_integration/cxg_immune_cell_atlas/train.h5ad",
     "output": "output.h5ad",
-    "model_name": "v2-medium",
+    "model_name": "medium-v1.5",
     "model": None,
     "predict_depth_mult": 1.0,
     "max_len": 12000,