remove torch import (#1995)

unit8co · Sep 16, 2023 · 74bd524 · 74bd524
1 parent 98c3ff4
commit 74bd524
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/darts/tests/models/forecasting/test_regression_ensemble_model.py b/darts/tests/models/forecasting/test_regression_ensemble_model.py
@@ -186,10 +186,11 @@ def test_train_n_points(self):
             ensemble.fit(self.combined)
 
         # using regression_train_n_point=-1 without pretraining
-        with pytest.raises(ValueError):
-            RegressionEnsembleModel(
-                self.get_global_models(), regression_train_n_points=-1
-            )
+        if TORCH_AVAILABLE:
+            with pytest.raises(ValueError):
+                RegressionEnsembleModel(
+                    self.get_global_models(), regression_train_n_points=-1
+                )
 
         # using regression_train_n_point=-1 with pretraining
         forecasting_models = [