chore: remove match statement to support python3.9 test run

aimclub · Feb 4, 2025 · 8363117 · 8363117
1 parent a57fb7a
commit 8363117
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/tests/unit/core/operation/interfaces/test_fedot_automl_evaluation_strategy.py b/tests/unit/core/operation/interfaces/test_fedot_automl_evaluation_strategy.py
@@ -21,13 +21,12 @@ def test_fedot_automl_strategy_fit_predict(task):
                            data_type=DataTypesEnum.table)
 
     params = OperationParameters(problem=task, timeout=0.1, n_jobs=1)
-    match task:
-        case 'classification':
-            strategy = FedotAutoMLClassificationStrategy(operation_type='fedot_cls', params=params)
-        case 'regression':
-            strategy = FedotAutoMLRegressionStrategy(operation_type='fedot_regr', params=params)
-        case _:
-            return
+    if task == 'classification':
+        strategy = FedotAutoMLClassificationStrategy(operation_type='fedot_cls', params=params)
+    elif task ==  'regression':
+        strategy = FedotAutoMLRegressionStrategy(operation_type='fedot_regr', params=params)
+    else:
+        return
 
     trained_operation = strategy.fit(input_data)