mend

mims-harvard · Oct 25, 2024 · 08db3c4 · 08db3c4
1 parent 91e5087
commit 08db3c4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tdc/test/test_model_server.py b/tdc/test/test_model_server.py
@@ -103,7 +103,8 @@ def testGeneformerTokenizer(self):
         geneformer = tdc_hf_interface("Geneformer")
         model = geneformer.load()
         tokenized_data = tokenizer.create_dataset(cells, metadata)
-        input_tensor = torch.squeeze(tokenized_data)
+        input_tensor = torch.tensor(tokenized_data)
+        input_tensor = torch.squeeze(input_tensor)
         try:
             out = model(tokenized_data)
         except Exception as e: