mend

mims-harvard · Jan 10, 2025 · a46a70f · a46a70f
1 parent 9e717a9
commit a46a70f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tdc/test/test_model_server.py b/tdc/test/test_model_server.py
@@ -97,6 +97,7 @@ def testscGPT(self):
         from tdc.multi_pred.anndata_dataset import DataLoader
         from tdc import tdc_hf_interface
         from tdc.model_server.tokenizers.scgpt import scGPTTokenizer
+        import torch
         adata = DataLoader("cellxgene_sample_small",
                            "./data",
                            dataset_names=["cellxgene_sample_small"],