Update preprocess_zzz.py

332plim · Jun 6, 2023 · 1a3545c · 1a3545c
1 parent ce3092a
commit 1a3545c
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/prepare/preprocess_zzz.py b/prepare/preprocess_zzz.py
@@ -1,3 +1,8 @@
+import os
+import sys
+current_dir = os.path.dirname(os.path.abspath(__file__))
+parent_dir = os.path.abspath(os.path.join(current_dir, os.pardir))
+sys.path.append(parent_dir)
 from tqdm import tqdm
 from torch.utils.data import DataLoader
 from omegaconf import OmegaConf