add current dir for model import

Author: jean2262

l2winddir/__init__.py

@@ -5,6 +5,4 @@
 
 from .data_module import *
 from .loss import *
-# from .main import *
-from .predict import *
-from .simple_conv import *
+from .predict import *

l2winddir/predict.py

@@ -7,7 +7,12 @@
 import torch
 import xarray as xr
 from omegaconf import OmegaConf
-import l2winddir.simple_conv
+import sys
+# Add the current directory to the Python path
+import os
+current_dir = os.path.dirname(os.path.abspath(__file__))
+sys.path.append(current_dir)
+
 import torch.multiprocessing as mp
 
 mp.set_start_method("spawn", force=True)