From 08db3c4d09aff60cae004bda54de6a30ab8692d5 Mon Sep 17 00:00:00 2001 From: Alejandro Velez-Arce Date: Fri, 25 Oct 2024 10:58:38 -0400 Subject: [PATCH] mend --- tdc/test/test_model_server.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tdc/test/test_model_server.py b/tdc/test/test_model_server.py index 880f2c87..bb0062e1 100644 --- 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: