mend

mims-harvard · Oct 23, 2024 · 3129e4c · 3129e4c
1 parent 8e851eb
commit 3129e4c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tdc/test/test_model_server.py b/tdc/test/test_model_server.py
@@ -102,7 +102,9 @@ def testGeneformerTokenizer(self):
         import torch
         geneformer = tdc_hf_interface("Geneformer")
         model = geneformer.load()
-        out = model(torch.tensor(cells).int())
+        input_tensor = torch.tensor(cells).int()
+        raise Exception("shape is", input_tensor.shape, "values are\n", input_tensor)
+        out = model(input_tensor)
         assert out, "FAILURE: Geneformer output is false-like. Value = {}".format(out)
         assert len(out) == len(cells), "FAILURE: Geneformer output and cells input don't have the same length. {} vs {}".format(len(out), len(cells))