fixes

src/model.py

@@ -1,7 +1,7 @@
 from darts import TimeSeries
 from darts.datasets import ILINetDataset
 from darts.metrics import mape
-from darts.model import ExponentialSmoothing
+from darts.models import ExponentialSmoothing
 from darts.utils.missing_values import fill_missing_values
 from darts.dataprocessing.transformers import Scaler
 import matplotlib.pyplot as plt
@@ -70,7 +70,7 @@ def scale_train(train_ili):
     return train_ili_scaled, scaler
 
 
-def train_(train_ili_scaled, val_ili, save_model_path='./models'):
+def train(train_ili_scaled):
     model = ExponentialSmoothing()
     model.fit(train_ili_scaled)
 
@@ -79,12 +79,11 @@ def train_(train_ili_scaled, val_ili, save_model_path='./models'):
 
 def save_model(model, path):
     model_name = str(model).split('(')[0]
-    path = f'{path}/{model_name}.pkl'
     if not os.path.exists(path=path):
         os.makedirs(path)
-        model.save(path)
-    else:
-        model = ExponentialSmoothing.load(path)
+        model.save(os.path.join(path, model_name))
+    #else:
+    #    model = ExponentialSmoothing.load(path)
 
 
 def load_model(path):