FFT_for_period

unit8co · Sep 26, 2024 · d9a0299 · d9a0299
1 parent 63b1358
commit d9a0299
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/darts/tests/models/forecasting/test_times_net_model.py b/darts/tests/models/forecasting/test_times_net_model.py
@@ -1,12 +1,12 @@
 import numpy as np
 import pandas as pd
 import pytest
+import torch
 
 from darts import TimeSeries
 from darts.models.forecasting.times_net_model import FFT_for_Period, TimesNetModel
 from darts.tests.conftest import TORCH_AVAILABLE, tfm_kwargs
 from darts.utils import timeseries_generation as tg
-import torch
 
 if not TORCH_AVAILABLE:
     pytest.skip(
@@ -122,7 +122,6 @@ def test_prediction_with_custom_encoders(self):
 
 
 class TestT_FFT_for_Period:
-
     sample_input = (
         torch.sin(torch.linspace(0, 4 * torch.pi, 100)).unsqueeze(0).unsqueeze(-1)
     )