mend

mims-harvard · Jan 7, 2025 · e6268fb · e6268fb
1 parent f6108df
commit e6268fb
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
@@ -109,7 +109,7 @@ def testscGPT(self):
         tokenized_data = tokenizer.tokenize_cell_vectors(
             adata.X.toarray(), gene_ids)
         first_embed = model(tokenized_data[0][1]).last_hidden_state
-        self.assertEqual(first_embed.shape[0], len(gene_ids))
+        self.assertEqual(first_embed.shape[0], len(tokenized_data[0][0]))
 
     def testGeneformerTokenizer(self):