updating the names of the checkpoints

src/methods/scprint/config.vsh.yaml

@@ -33,23 +33,23 @@ info:
   method_types: [embedding]
   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"
   test_setup:
     run:
-      model_name: small
+      model_name: small-v1
       batch_size: 16
       max_len: 100
 
 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-v1.5", "small-v1"]
+    default: "medium-v1.5"
   - name: --model
     type: file
     description: Path to the scPRINT model.
@@ -61,7 +61,7 @@ arguments:
   - name: --max_len
     type: integer
     description: The maximum length of the gene sequence.
-    default: 4000
+    default: 2300
 
 resources:
   - type: python_script

src/methods/scprint/script.py

@@ -14,7 +14,7 @@
 par = {
     "input": "resources_test/task_batch_integration/cxg_immune_cell_atlas/dataset.h5ad",
     "output": "output.h5ad",
-    "model_name": "v2-medium",
+    "model_name": "medium-v1.5",
     "model": None,
 }
 meta = {"name": "scprint"}