mend

mims-harvard · Jan 21, 2025 · 79b6030 · 79b6030
1 parent bd255e1
commit 79b6030
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,