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))