Fix

aimclub · Dec 28, 2023 · d764566 · d764566
1 parent 1266211
commit d764566
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/test/integration/api/test_api_utils.py b/test/integration/api/test_api_utils.py
@@ -5,14 +5,15 @@
 
 from examples.simple.classification.classification_pipelines import (classification_pipeline_with_balancing,
                                                                      classification_pipeline_without_balancing)
-from fedot.api.api_utils.assumptions.assumptions_builder import AssumptionsBuilder
 from fedot import Fedot
+from fedot.api.api_utils.assumptions.assumptions_builder import AssumptionsBuilder
 from fedot.core.data.data_split import train_test_data_setup
 from fedot.core.pipelines.node import PipelineNode
 from fedot.core.pipelines.pipeline import Pipeline
 from fedot.core.repository.tasks import Task, TaskTypesEnum, TsForecastingParams
 from fedot.preprocessing.preprocessing import DataPreprocessor
-from test.integration.api.test_main_api import get_dataset, get_cholesterol_dataset
+from test.data.datasets import get_cholesterol_dataset
+from test.integration.api.test_main_api import get_dataset
 from test.unit.tasks.test_classification import get_binary_classification_data