inf condition simplification

aimclub · Jul 31, 2023 · 9f39d7b · 9f39d7b
1 parent a4f23d1
commit 9f39d7b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fedot/core/data/data_preprocessing.py b/fedot/core/data/data_preprocessing.py
@@ -14,7 +14,7 @@ def data_type_is_suitable_for_preprocessing(data: InputData) -> bool:
 
 def replace_inf_with_nans(input_data: InputData):
     features = input_data.features
-    is_inf = (features == np.inf) | (features == -np.inf)
+    is_inf = np.isin(features, [np.inf, -np.inf])
     if np.any(is_inf):
         features[is_inf] = np.nan