Fix/failing unit tests (#2420)

* fix backtest unit tests * fix custom regression model
unit8co · Jun 19, 2024 · f7d38c3 · f7d38c3
1 parent ad150e9
commit f7d38c3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git a/darts/tests/models/forecasting/test_backtesting.py b/darts/tests/models/forecasting/test_backtesting.py
@@ -14,8 +14,10 @@
     ARIMA,
     FFT,
     ExponentialSmoothing,
+    LinearRegressionModel,
     NaiveDrift,
     NaiveSeasonal,
+    RandomForest,
     Theta,
 )
 from darts.tests.conftest import TORCH_AVAILABLE, tfm_kwargs
@@ -29,12 +31,7 @@
 
 
 if TORCH_AVAILABLE:
-    from darts.models import (
-        BlockRNNModel,
-        LinearRegressionModel,
-        RandomForest,
-        TCNModel,
-    )
+    from darts.models import BlockRNNModel, TCNModel
 
 
 def get_dummy_series(

diff --git a/examples/20-RegressionModel-examples.ipynb b/examples/20-RegressionModel-examples.ipynb
@@ -976,7 +976,7 @@
     "        self.weights = weights\n",
     "        self.norm_coef = sum(weights)\n",
     "\n",
-    "    def fit(self, X: np.ndarray, y: np.ndarray):\n",
+    "    def fit(self, X: np.ndarray, y: np.ndarray, *args, **kwargs):\n",
     "        return self\n",
     "\n",
     "    def predict(self, X: np.ndarray):\n",