Conditional import of GaussianLikelihood in torch tests

unit8co · Jun 18, 2022 · ebab32c · ebab32c
1 parent c8169ec
commit ebab32c
Showing 1 changed file with 1 addition 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
@@ -8,7 +8,6 @@
 from darts import TimeSeries
 from darts.logging import get_logger
 from darts.tests.base_test_class import DartsBaseTestClass
-from darts.utils.likelihood_models import GaussianLikelihood
 
 logger = get_logger(__name__)
 
@@ -21,6 +20,7 @@
     )
 
     from darts.models.forecasting.rnn_model import RNNModel
+    from darts.utils.likelihood_models import GaussianLikelihood
 
     TORCH_AVAILABLE = True
 except ImportError: