diff --git a/tdc/model_server/tokenizers/geneformer.py b/tdc/model_server/tokenizers/geneformer.py index 3fa338cd..48225682 100644 --- a/tdc/model_server/tokenizers/geneformer.py +++ b/tdc/model_server/tokenizers/geneformer.py @@ -1,11 +1,10 @@ import numpy as np import scipy.sparse as sp -from geneformer import TranscriptomeTokenizer from ...utils.load import pd_load, download_wrapper -class GeneformerTokenizer(TranscriptomeTokenizer): +class GeneformerTokenizer: """ Uses Geneformer Utils to parse zero-shot model server requests for tokenizing single-cell gene expression data. diff --git a/tdc/test/test_model_server.py b/tdc/test/test_model_server.py index 82a10f8a..3ccd56c0 100644 --- a/tdc/test/test_model_server.py +++ b/tdc/test/test_model_server.py @@ -95,7 +95,7 @@ def testGeneformerTokenizer(self): assert x[0] # test Geneformer can serve the request - cells, metadata = x + cells, _ = x assert cells, "FAILURE: cells false-like. Value is = {}".format(cells) assert len(cells) > 0, "FAILURE: length of cells <= 0 {}".format(cells) from tdc import tdc_hf_interface