fixed the test workwflow

.github/workflows/test_linux_custom_dataloader.yml

@@ -66,7 +66,7 @@ jobs:
           GH_TOKEN: ${{ secrets.GH_TOKEN }}
         run: |
           git config --global url."https://${GH_TOKEN}:[email protected]/".insteadOf "https://github.com/"
-          git clone --single-branch --branch ebezzi/census-scvi-datamodule https://github.com/chanzuckerberg/cellxgene-census.git
+          git clone --single-branch --branch ebezzi/census-scvi-datamodule https://github.com/ori-kron-wis/cellxgene-census.git
           git clone --single-branch --branch main https://github.com/jkobject/scDataLoader.git
 
       - name: Run specific custom dataloader pytest

tests/dataloaders/test_czi_custom_dataloader.py

@@ -2,20 +2,31 @@
 
 from pprint import pprint
 
-import cellxgene_census
 import numpy as np
 import pandas as pd
 import pytest
-import tiledbsoma as soma
-from cellxgene_census.experimental.ml import experiment_dataloader
-from cellxgene_census.experimental.ml.datamodule import CensusSCVIDataModule
 
 import scvi
 from scvi.data import synthetic_iid
 
 
 @pytest.mark.custom_dataloader
-def test_czi_custom_dataloader(save_path="."):
+def test_czi_custom_dataloader(save_path):
+    # local bracnh with fix only for this test
+    import sys
+
+    # should be ready for importing the cloned branch on a remote machine that runs github action
+    sys.path.insert(
+        0,
+        "/home/runner/work/scvi-tools/scvi-tools/"
+        "cellxgene-census/api/python/cellxgene_census/src",
+    )
+    sys.path.insert(0, "src")
+    import cellxgene_census
+    import tiledbsoma as soma
+    from cellxgene_census.experimental.ml import experiment_dataloader
+    from cellxgene_census.experimental.ml.datamodule import CensusSCVIDataModule
+
     # this test checks the local custom dataloder made by CZI and run several tests with it
     census = cellxgene_census.open_soma(census_version="stable")