fix: fix checking classification extras

tinkoff-ai · Apr 27, 2023 · f87c9fa · f87c9fa
1 parent ce48ab8
commit f87c9fa
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/etna/experimental/classification/__init__.py b/etna/experimental/classification/__init__.py
@@ -1,2 +1,5 @@
-from etna.experimental.classification.classification import TimeSeriesBinaryClassifier
-from etna.experimental.classification.predictability import PredictabilityAnalyzer
+from etna import SETTINGS
+
+if SETTINGS.classification_required:
+    from etna.experimental.classification.classification import TimeSeriesBinaryClassifier
+    from etna.experimental.classification.predictability import PredictabilityAnalyzer
diff --git a/etna/experimental/classification/feature_extraction/__init__.py b/etna/experimental/classification/feature_extraction/__init__.py
@@ -1,3 +1,6 @@
-from etna.experimental.classification.feature_extraction.base import BaseTimeSeriesFeatureExtractor
-from etna.experimental.classification.feature_extraction.tsfresh import TSFreshFeatureExtractor
-from etna.experimental.classification.feature_extraction.weasel import WEASELFeatureExtractor
+from etna import SETTINGS
+
+if SETTINGS.classification_required:
+    from etna.experimental.classification.feature_extraction.base import BaseTimeSeriesFeatureExtractor
+    from etna.experimental.classification.feature_extraction.tsfresh import TSFreshFeatureExtractor
+    from etna.experimental.classification.feature_extraction.weasel import WEASELFeatureExtractor