Add use_lmdb config option.

usnistgov · Apr 30, 2024 · 3f7ada2 · 3f7ada2
1 parent 4f6f871
commit 3f7ada2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/alignn/pretrained.py b/alignn/pretrained.py
@@ -6,7 +6,7 @@
 import zipfile
 from tqdm import tqdm
 from alignn.models.alignn import ALIGNN, ALIGNNConfig
-from alignn.data import get_torch_dataset
+from alignn.dataset import get_torch_dataset
 from torch.utils.data import DataLoader
 import tempfile
 import torch