From b5ae7c61b19dd771e0d28df2325126e9294744f9 Mon Sep 17 00:00:00 2001 From: macmini-claw Date: Wed, 12 Aug 2026 08:53:10 +0200 Subject: [PATCH] fix(scprint): update pretrained checkpoint names --- src/methods/scprint/config.vsh.yaml | 10 +++++----- src/methods/scprint/script.py | 2 +- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/src/methods/scprint/config.vsh.yaml b/src/methods/scprint/config.vsh.yaml index 40af7c8..96fb09a 100644 --- a/src/methods/scprint/config.vsh.yaml +++ b/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-v1.5", "small-v1"] + default: "medium-v1.5" - name: --model type: file description: Path to the scPRINT model. diff --git a/src/methods/scprint/script.py b/src/methods/scprint/script.py index b230801..a5998b5 100644 --- a/src/methods/scprint/script.py +++ b/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,