only pass the input ids

mims-harvard · Jan 9, 2025 · c1d800c · c1d800c
1 parent a41cd0e
commit c1d800c
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tdc/test/test_model_server.py b/tdc/test/test_model_server.py
@@ -110,11 +110,9 @@ def testscGPT(self):
             adata.X.toarray(), gene_ids)
         mask = [x != 0 for x in tokenized_data[0][1]]
         assert sum(mask) != 0, "FAILURE: mask is empty"
-        first_embed = model(tokenized_data[0][1],
-                            input_ids=tokenized_data[0][0],
+        first_embed = model(tokenized_data[0][0],
                             attention_mask=mask).last_hidden_state
         print(f"scgpt ran successfully. here is an output {first_embed}")
-        self.assertEqual(first_embed.shape[0], len(tokenized_data[0][0]))
 
     def testGeneformerTokenizer(self):