Fix imports in main.py

main.py

@@ -3,6 +3,13 @@
 ##       ANY CHANGES HERE WILL NOT BE REFLECTED IN THE EVALUATION SETUP
 ##       PLEASE MAKE YOUR CHANGES IN `run.py`.
 ##
+
+import numpy as np
+from tqdm.auto import tqdm
+
+from dataset import ZEWDPCBaseDataset, ZEWDPCProtectedDataset
+from run import ZEWDPCBaseRun
+
 ####################################################################################
 ####################################################################################
 ##