mend

.github/workflows/conda-tests.yml

@@ -48,6 +48,7 @@ jobs:
           echo "Creating Conda Environment from environment.yml"
           conda env create -f environment.yml
           conda activate tdc-conda-env
+          python run_tests.py tdc.test.test_model_server.TestModelServer.testscVI 
           python run_tests.py
           yapf --style=google -r -d tdc
           conda deactivate

tdc/test/test_model_server.py

@@ -31,8 +31,18 @@ def setUp(self):
         print(os.getcwd())
         self.resource = cellxgene_census.CensusResource()
 
-    def testFake(self):
-        print("success")
+    def testscVI(self):
+        from tdc.multi_pred.anndata_dataset import DataLoader
+        from tdc import tdc_hf_interface
+        # run scvi test
+        adata = DataLoader("cellxgene_sample_small",
+                    "./data",
+                    dataset_names=["cellxgene_sample_small"],
+                    no_convert=True).adata
+        # run scVI on a single sample
+        scvi = tdc_hf_interface("scVI")
+        model = scvi.load()  # this line can cause segmentation fault
+        pass
 
     def testscGPT(self):
         from tdc.multi_pred.anndata_dataset import DataLoader