fix the c.dataloders test

scverse · Oct 10, 2024 · ed77a65 · ed77a65
1 parent 1801604
commit ed77a65
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/.github/workflows/test_linux_custom_dataloader.yml b/.github/workflows/test_linux_custom_dataloader.yml
@@ -35,7 +35,7 @@ jobs:
       fail-fast: false
       matrix:
         os: [ubuntu-latest]
-        python: ["3.11"]
+        python: ["3.10"]
 
     name: integration
 
@@ -60,6 +60,8 @@ jobs:
           python -m pip install s3fs
           python -m pip install torchdata
           python -m pip install psutil
+          python -m pip install cellxgene-census
+          python -m pip install lamindb
 
       - name: Install Specific Branch of Repository
         env:

diff --git a/tests/dataloaders/test_custom_dataloader.py b/tests/dataloaders/test_custom_dataloader.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import os
 from pprint import pprint
 
@@ -11,7 +13,7 @@
 
 
 @pytest.mark.custom_dataloader
-def test_czi_custom_dataloader(save_path="."):
+def test_czi_custom_dataloader(save_path):
     # local branch with fix only for this test
     import sys
 
@@ -235,7 +237,7 @@ def test_czi_custom_dataloader(save_path="."):
 
 
 @pytest.mark.custom_dataloader
-def test_lamindb_custom_dataloader(save_path="."):
+def test_lamindb_custom_dataloader(save_path):
     # initialize a local lamin database
     os.system("lamin init --storage ~/scdataloader2 --schema bionty")
     # os.system("lamin close")