move torch import in tests for flavors

unit8co · Feb 16, 2022 · a6ce1d2 · a6ce1d2
1 parent 31db666
commit a6ce1d2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/darts/tests/models/forecasting/test_torch_forecasting_model.py b/darts/tests/models/forecasting/test_torch_forecasting_model.py
@@ -4,7 +4,6 @@
 from unittest.mock import patch
 
 import pandas as pd
-import torch
 
 from darts import TimeSeries
 from darts.logging import get_logger
@@ -13,6 +12,8 @@
 logger = get_logger(__name__)
 
 try:
+    import torch
+
     from darts.models.forecasting.rnn_model import RNNModel
 
     TORCH_AVAILABLE = True