datasets load necessary libs and change imports in tests

rctn · Nov 22, 2024 · 61cf6c2 · 61cf6c2
1 parent 32848c3
commit 61cf6c2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/sparsecoding/datasets.py b/sparsecoding/datasets.py
@@ -1,4 +1,7 @@
 import torch
+import os
+from scipy.io import loadmat
+from sparsecoding.transforms.patch import patchify
 from torch.utils.data import Dataset
 
 from sparsecoding.priors import Prior

diff --git a/tests/inference/common.py b/tests/inference/common.py
@@ -1,7 +1,7 @@
 import torch
 
 from sparsecoding.priors import L0Prior, SpikeSlabPrior
-from sparsecoding.data.datasets.bars import BarsDataset
+from sparsecoding.datasets import BarsDataset
 
 torch.manual_seed(1997)