Skip to content

Commit

Permalink
remove geneformer package dependency
Browse files Browse the repository at this point in the history
  • Loading branch information
amva13 committed Oct 26, 2024
1 parent 01e6eb5 commit b913742
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 3 deletions.
3 changes: 1 addition & 2 deletions tdc/model_server/tokenizers/geneformer.py
Original file line number Diff line number Diff line change
@@ -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.
Expand Down
2 changes: 1 addition & 1 deletion tdc/test/test_model_server.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit b913742

Please sign in to comment.