fixing that 3

AIStream-Peelout · Oct 23, 2023 · dd8b04c · dd8b04c
1 parent f3864a8
commit dd8b04c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/flood_forecast/basic/base_line_methods.py b/flood_forecast/basic/base_line_methods.py
@@ -3,8 +3,7 @@
 
 class NaiveBase(torch.nn.Module):
     """
-    A very simple baseline model that returns
-    the fixed value based on the input sequence.
+    A very simple baseline model that returns the fixed value based on the input sequence.
     No learning used at all.
     """
 

diff --git a/flood_forecast/preprocessing/pytorch_loaders.py b/flood_forecast/preprocessing/pytorch_loaders.py
@@ -661,8 +661,8 @@ def __init__(self, series_id_col: str, main_params: dict, return_method: str, re
         :param forecast_total: _description_, defaults to 336
         :type forecast_total: int, optional
         """
-        CSVSeriesIDLoader().__init__(self, series_id_col, main_params, return_method, return_all)
-        CSVTestLoader().__init__(self, df_path="", forecast_total=forecast_total, )
+        super.__init__(series_id_col=series_id_col, main_params=main_params, return_method=return_method,
+                       return_all=return_all)
         self.forecast_total = forecast_total
 
     def get_from_start_date_all(self, forecast_start: datetime, series_id: int = None):