From 79b6030d925b3c75c456c39abf7ff08b62cc3024 Mon Sep 17 00:00:00 2001 From: Alejandro Velez Date: Tue, 21 Jan 2025 08:04:31 -0500 Subject: [PATCH] mend --- tdc/test/test_model_server.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tdc/test/test_model_server.py b/tdc/test/test_model_server.py index d57a7116..77dc46cd 100644 --- a/tdc/test/test_model_server.py +++ b/tdc/test/test_model_server.py @@ -60,7 +60,7 @@ def testGeneformerPerturb(self): dataset = "scperturb_drug_AissaBenevolenskaya2021" data = PerturbOutcome(dataset) adata = data.adata - tokenizer = GeneformerTokenizer() + tokenizer = GeneformerTokenizer(max_input_size=3) adata.var["feature_id"] = adata.var.index.map( lambda x: tokenizer.gene_name_id_dict.get(x, 0)) x = tokenizer.tokenize_cell_vectors(adata,