fixed another import error

scverse · Dec 11, 2024 · ce55660 · ce55660
1 parent e889765
commit ce55660
Showing 1 changed file with 3 additions and 8 deletions.
diff --git a/src/scvi/data/_anntorchdataset.py b/src/scvi/data/_anntorchdataset.py
@@ -6,14 +6,7 @@
 import h5py
 import numpy as np
 import pandas as pd
-
-try:
-    from anndata.abc import CSCDataset, CSRDataset
-
-    SparseDataset = (CSRDataset, CSCDataset)
-except ImportError:
-    from anndata._core.sparse_dataset import SparseDataset
-
+from anndata.abc import CSCDataset, CSRDataset
 from scipy.sparse import issparse
 from torch.utils.data import Dataset
 
@@ -27,6 +20,8 @@
 
 logger = logging.getLogger(__name__)
 
+SparseDataset = (CSRDataset, CSCDataset)
+
 
 class AnnTorchDataset(Dataset):
     """Extension of :class:`~torch.utils.data.Dataset` for :class:`~anndata.AnnData` objects.