correct import

skothenhill-nv · skothenhill-nv · commit 2eacb1236c9a · 2025-08-15T09:34:32.000-07:00
Signed-off-by: Steven &lt;skothenhill@nvidia.com&gt;
diff --git a/sub-packages/bionemo-geneformer/tests/bionemo/geneformer/test_block_sampling.py b/sub-packages/bionemo-geneformer/tests/bionemo/geneformer/test_block_sampling.py
@@ -18,7 +18,7 @@
 import pytest
 from torch.utils.data import Dataset
 
-from bionemo.geneformer.data.singlecell.block_sampling import MapStyleScDataset, scDataset
+from bionemo.geneformer.data.block_sampling import MapStyleScDataset, scDataset
 
 
 class DummyDataset(Dataset):
