From 0b37be8fb8c3747787cdb2354c03b6d431fd90e3 Mon Sep 17 00:00:00 2001
From: Alejandro Velez-Arce <alex@alexv24.com>
Date: Mon, 21 Oct 2024 20:52:07 -0400
Subject: [PATCH] geneformer tokenizer fails on scperturb

---
 tdc/test/test_model_server.py | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/tdc/test/test_model_server.py b/tdc/test/test_model_server.py
index 70c93459..d26f4041 100644
--- a/tdc/test/test_model_server.py
+++ b/tdc/test/test_model_server.py
@@ -100,7 +100,21 @@ def testGeneformerTokenizer(self):
         tokenizer = GeneformerTokenizer()
         print("testing tokenizer")
         x = tokenizer.tokenize_cell_vectors(adata)
-        assert x
+        assert x[0]
+
+        # test Geneformer can serve the request
+        cells = x[0]
+        print("cells is", len(cells), cells)
+        assert cells[0]
+        assert len(cells[0]) > 0
+        from tdc import tdc_hf_interface
+        import torch
+        geneformer = tdc_hf_interface("Geneformer")
+        model = geneformer.load()
+        out = model(torch.tensor(cells))
+        assert out
+        assert out[0]
+        assert len(out[0]) > 0
 
     def tearDown(self):
         try: