Welding/WeldingSpotPerformance
9
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases 2 Wiki Activity

fix: код изменен для поддержки будущих версий pandas

Browse Source
This commit is contained in:
Andrew 2025-02-10 14:55:10 +03:00
parent ec8ce9b59a
commit dea6d9899c
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/controller/converter.py
View File

@ -13,10 +13,11 @@ class DataConverter(BaseDataConverter):
def _replace_bool(dataframe: pd.DataFrame) -> pd.DataFrame: def _replace_bool(dataframe: pd.DataFrame) -> pd.DataFrame:
try: try:
bool_columns = dataframe.columns[dataframe.isin([True, False]).all()] 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 return dataframe
except AttributeError as e: except AttributeError as e:
logger.error(f"_replace_bool - AttributeError: Проверьте, что переданный объект является DataFrame. {e}") logger.warning(f"_replace_bool - AttributeError: Проверьте, что переданный объект является DataFrame. {e}")
return None return None
except (TypeError, ValueError) as e: except (TypeError, ValueError) as e:
logger.error(f"_replace_bool - Ошибка типа или значения: {e}") logger.error(f"_replace_bool - Ошибка типа или значения: {e}")
@ -46,7 +47,7 @@ class DataConverter(BaseDataConverter):
dataframe = dataframe.loc[:, ~dataframe.columns.duplicated()] dataframe = dataframe.loc[:, ~dataframe.columns.duplicated()]
return dataframe return dataframe
except AttributeError as e: except AttributeError as e:
logger.error(f"_fix_headers - AttributeError: Проверьте, что переданный объект является DataFrame. {e}") logger.warning(f"_fix_headers - AttributeError: Проверьте, что переданный объект является DataFrame. {e}")
return None return None
except Exception as e: except Exception as e:
logger.error(f"_fix_headers - Непредвиденная ошибка: {e}") logger.error(f"_fix_headers - Непредвиденная ошибка: {e}")