Imports cleanup in model tests.

sarusso · Nov 3, 2024 · 6eaebe6 · 6eaebe6
1 parent ad2a576
commit 6eaebe6
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 18 deletions.
diff --git a/timeseria/tests/test_models_anomaly_detectors.py b/timeseria/tests/test_models_anomaly_detectors.py
@@ -2,8 +2,8 @@
 import os
 import tempfile
 import random
-import numpy
 from math import sin, cos
+
 from ..datastructures import TimePoint, DataTimeSlot, DataTimePoint, TimeSeries
 from ..models.forecasters import PeriodicAverageForecaster
 from ..models.reconstructors import PeriodicAverageReconstructor

diff --git a/timeseria/tests/test_models_base.py b/timeseria/tests/test_models_base.py
@@ -1,16 +1,11 @@
 import unittest
 import os
 import tempfile
-from math import sin, cos
 from propertime.utils import dt
 
 from ..datastructures import TimePoint, DataTimeSlot, DataTimePoint, TimeSeries
 from ..models.base import Model, _KerasModel
-from ..models.reconstructors import PeriodicAverageReconstructor, ProphetReconstructor
-from ..models.forecasters import ProphetForecaster, PeriodicAverageForecaster, ARIMAForecaster, AARIMAForecaster, LSTMForecaster
-from ..models.anomaly_detectors import PeriodicAverageAnomalyDetector
-from ..exceptions import NotFittedError, NonContiguityError
-from ..storages import CSVFileStorage
+from ..exceptions import NotFittedError
 
 # Setup logging
 from .. import logger

diff --git a/timeseria/tests/test_models_forecasters.py b/timeseria/tests/test_models_forecasters.py
@@ -1,17 +1,12 @@
 import unittest
 import os
 import tempfile
-import random
-import numpy
 from math import sin, cos
 from propertime.utils import dt
 
 from ..datastructures import TimePoint, DataTimeSlot, DataTimePoint, TimeSeries
-from ..models.base import Model, _KerasModel
-from ..models.reconstructors import PeriodicAverageReconstructor, ProphetReconstructor
 from ..models.forecasters import ProphetForecaster, PeriodicAverageForecaster, ARIMAForecaster, AARIMAForecaster, LSTMForecaster
-from ..models.anomaly_detectors import PeriodicAverageAnomalyDetector
-from ..exceptions import NotFittedError, NonContiguityError
+from ..exceptions import NonContiguityError
 from ..storages import CSVFileStorage
 from .. import TEST_DATASETS_PATH
 

diff --git a/timeseria/tests/test_models_reconstructors.py b/timeseria/tests/test_models_reconstructors.py
@@ -1,15 +1,10 @@
 import unittest
 import os
 import tempfile
-from math import sin, cos
 from propertime.utils import dt
 
 from ..datastructures import TimePoint, DataTimeSlot, DataTimePoint, TimeSeries
-from ..models.base import Model, _KerasModel
 from ..models.reconstructors import PeriodicAverageReconstructor, ProphetReconstructor, LinearInterpolationReconstructor
-from ..models.forecasters import ProphetForecaster, PeriodicAverageForecaster, ARIMAForecaster, AARIMAForecaster, LSTMForecaster
-from ..models.anomaly_detectors import PeriodicAverageAnomalyDetector
-from ..exceptions import NotFittedError, NonContiguityError
 from ..storages import CSVFileStorage
 
 # Setup logging