Fix : voxdataset import error

Make-Zenerator · Mar 13, 2024 · 4c93d4f · 4c93d4f
1 parent 16bf2e3
commit 4c93d4f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/serving/sf2f/datasets/build_dataset.py b/serving/sf2f/datasets/build_dataset.py
@@ -2,7 +2,10 @@
 import os
 import os.path as osp
 import numpy as np
-from datasets import VoxDataset
+try: 
+    from .datasets import VoxDataset
+except:
+    from ..dataset import VoxDataset
 from torch.utils.data import DataLoader