diff --git a/src/controller/converter.py b/src/controller/converter.py index 343576c..5367ede 100644 --- a/src/controller/converter.py +++ b/src/controller/converter.py @@ -13,10 +13,11 @@ class DataConverter(BaseDataConverter): def _replace_bool(dataframe: pd.DataFrame) -> pd.DataFrame: try: bool_columns = dataframe.columns[dataframe.isin([True, False]).all()] - dataframe.loc[:, bool_columns] = dataframe.loc[:, bool_columns].replace({True: 1, False: 0}) + dataframe = dataframe.astype({col: int for col in bool_columns}) + return dataframe except AttributeError as e: - logger.error(f"_replace_bool - AttributeError: Проверьте, что переданный объект является DataFrame. {e}") + logger.warning(f"_replace_bool - AttributeError: Проверьте, что переданный объект является DataFrame. {e}") return None except (TypeError, ValueError) as e: logger.error(f"_replace_bool - Ошибка типа или значения: {e}") @@ -46,7 +47,7 @@ class DataConverter(BaseDataConverter): dataframe = dataframe.loc[:, ~dataframe.columns.duplicated()] return dataframe except AttributeError as e: - logger.error(f"_fix_headers - AttributeError: Проверьте, что переданный объект является DataFrame. {e}") + logger.warning(f"_fix_headers - AttributeError: Проверьте, что переданный объект является DataFrame. {e}") return None except Exception as e: logger.error(f"_fix_headers - Непредвиденная ошибка: {e}")