From e07347de7222ee1e0ab2dbf7142cac3fd510e1f6 Mon Sep 17 00:00:00 2001
From: Lopa10ko <justforprojects@mail.ru>
Date: Mon, 3 Feb 2025 17:53:26 +0300
Subject: [PATCH] chore: remove federated_automl from strategies test

---
 .../unit/api/utils/test_industrial_strategy.py | 18 ++++++++++--------
 1 file changed, 10 insertions(+), 8 deletions(-)

diff --git a/tests/unit/api/utils/test_industrial_strategy.py b/tests/unit/api/utils/test_industrial_strategy.py
index 10a93c03..27d18287 100644
--- a/tests/unit/api/utils/test_industrial_strategy.py
+++ b/tests/unit/api/utils/test_industrial_strategy.py
@@ -11,12 +11,13 @@
 INDUSTRIAL_PARAMS = get_industrial_params()
 DEFAULT_AUTOML_LEARNING_CONFIG['timeout'] = 0.1
 
-CONFIGS = {'federated_automl': {'industrial_config': {'problem': 'classification',
-                                                      'strategy': 'federated_automl',
-                                                      'strategy_params': INDUSTRIAL_PARAMS},
-                                'learning_config': DEFAULT_CLF_LEARNING_CONFIG,
-                                'automl_config': DEFAULT_CLF_AUTOML_CONFIG,
-                                'compute_config': DEFAULT_COMPUTE_CONFIG},
+CONFIGS = {
+           # 'federated_automl': {'industrial_config': {'problem': 'classification',
+           #                                            'strategy': 'federated_automl',
+           #                                            'strategy_params': INDUSTRIAL_PARAMS},
+           #                      'learning_config': DEFAULT_CLF_LEARNING_CONFIG,
+           #                      'automl_config': DEFAULT_CLF_AUTOML_CONFIG,
+           #                      'compute_config': DEFAULT_COMPUTE_CONFIG},
 
            'lora_strategy': {'industrial_config': {'problem': 'classification',
                                                    'strategy': 'lora_strategy',
@@ -42,14 +43,15 @@
            #                             'compute_config': DEFAULT_COMPUTE_CONFIG},
 
            # 'forecasting_exogenous': {}
-           }
+}
 
 
 @pytest.mark.parametrize('strategy', STRATEGY)
 def test_custom_strategy(strategy):
     if strategy in CONFIGS.keys():
         cnfg = CONFIGS[strategy]
-        train_data, test_data = get_data_by_task(cnfg['industrial_config']['problem'])
+        train_data, test_data = map(lambda x: (x[0].values, x[1]),
+                                    get_data_by_task(cnfg['industrial_config']['problem']))
         n_samples = train_data[0].shape[0]
 
         industrial = FedotIndustrial(**cnfg)