From e72d938f2f869b8006f85dacba921d61ff057843 Mon Sep 17 00:00:00 2001 From: David Chang Date: Mon, 26 Feb 2024 10:16:58 +0800 Subject: [PATCH] ver Feb26th fixed bug caused by new-added NoneType --- desktop_env/evaluators/metrics/table.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/desktop_env/evaluators/metrics/table.py b/desktop_env/evaluators/metrics/table.py index e3dc881..24e098e 100644 --- a/desktop_env/evaluators/metrics/table.py +++ b/desktop_env/evaluators/metrics/table.py @@ -153,10 +153,13 @@ def compare_table(result: str, expected: str = None, **options) -> float: # precision: int as number of decimal digits, default to 4 error_limit: int = r.get("precision", 4) - sheet1: pd.DataFrame = _load_sheet(*parse_idx(r["sheet_idx0"], pdworkbookr, pdworkbooke)).round(error_limit) + sheet1: pd.DataFrame = _load_sheet(*parse_idx(r["sheet_idx0"], pdworkbookr, pdworkbooke)) if sheet1 is None: return 0. - sheet2: pd.DataFrame = _load_sheet(*parse_idx(r["sheet_idx1"], pdworkbookr, pdworkbooke)).round(error_limit) + sheet2: pd.DataFrame = _load_sheet(*parse_idx(r["sheet_idx1"], pdworkbookr, pdworkbooke)) + + sheet1 = sheet1.round() + sheet2 = sheet2.round() metric: bool = sheet1.equals(sheet2) logger.debug("Sheet1: \n%s", str(sheet1)) logger.debug("Sheet2: \n%s", str(sheet2))