From a37f8a39e01ba550bc3e5fdefb8a065747924d62 Mon Sep 17 00:00:00 2001 From: Timothyxxx <384084775@qq.com> Date: Tue, 12 Dec 2023 18:12:37 +0800 Subject: [PATCH] Initialize evaluation protocols and examples; Implement one kind of eval; Update requirements --- desktop_env/evaluators/__init__.py | 5 +++++ desktop_env/evaluators/replay.py | 0 desktop_env/evaluators/table.py | 14 ++++++++++++++ 3 files changed, 19 insertions(+) create mode 100644 desktop_env/evaluators/__init__.py create mode 100644 desktop_env/evaluators/replay.py create mode 100644 desktop_env/evaluators/table.py diff --git a/desktop_env/evaluators/__init__.py b/desktop_env/evaluators/__init__.py new file mode 100644 index 0000000..32a19e7 --- /dev/null +++ b/desktop_env/evaluators/__init__.py @@ -0,0 +1,5 @@ +from .table import compare_table + +eval_funcs = { + "compare_table(expected, actual)": compare_table +} diff --git a/desktop_env/evaluators/replay.py b/desktop_env/evaluators/replay.py new file mode 100644 index 0000000..e69de29 diff --git a/desktop_env/evaluators/table.py b/desktop_env/evaluators/table.py new file mode 100644 index 0000000..bd49297 --- /dev/null +++ b/desktop_env/evaluators/table.py @@ -0,0 +1,14 @@ +def compare_table(expected, actual): + import pandas as pd + df1 = pd.read_excel(expected) + df2 = pd.read_excel(actual) + + # Compare the DataFrames + return 1 if df1.equals(df2) else 0 + + +if __name__ == '__main__': + path1 = "" + path2 = "" + + print(compare_table(path1, path2))