pandas version fix

aimclub · Jul 31, 2023 · a4f23d1 · a4f23d1
1 parent d75a7ab
commit a4f23d1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fedot/preprocessing/data_types.py b/fedot/preprocessing/data_types.py
@@ -362,7 +362,7 @@ def define_column_types(table: Optional[np.ndarray]) -> pd.DataFrame:
     types, which column contains.
     """
     table_of_types = pd.DataFrame(table, copy=True)
-    table_of_types = table_of_types.replace(np.nan, None).applymap(lambda el: TYPE_TO_ID[type(el)])
+    table_of_types = table_of_types.replace({np.nan: None}).applymap(lambda el: TYPE_TO_ID[type(el)])
 
     # Build dataframe with unique types for each column
     uniques = table_of_types.apply(pd.unique, result_type='reduce').to_frame(_TYPES).T