Fix test_regression_data_operations with inf data after poly_features

aimclub · Aug 21, 2024 · 3544636 · 3544636
1 parent 5cf8d1b
commit 3544636
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/fedot/core/operations/evaluation/operation_implementations/implementation_interfaces.py b/fedot/core/operations/evaluation/operation_implementations/implementation_interfaces.py
@@ -107,6 +107,8 @@ def transform(self, input_data: InputData) -> OutputData:
         else:
             transformed_features = features
 
+        transformed_features = np.nan_to_num(transformed_features, copy=False, nan=0, posinf=0, neginf=0)
+
         # Update features and column types
         output_data = self._convert_to_output(input_data, transformed_features)
         self._update_column_types(source_features_shape, output_data)