You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

custom_tabulate.py 809 B

1234567891011121314151617181920212223242526272829303132333435
  1. import os
  2. import sys
  3. import re
  4. from agbench.tabulate_cmd import default_tabulate
  5. def scorer(instance_dir):
  6. # Read the console
  7. console_log_file = os.path.join(instance_dir, "console_log.txt")
  8. if not os.path.isfile(console_log_file):
  9. return None
  10. console_log = ""
  11. with open(console_log_file, "rt") as fh:
  12. console_log = fh.read()
  13. final_score = None
  14. m = re.search(r"FINAL SCORE:(.*?)\n", console_log, re.DOTALL)
  15. if m:
  16. final_score = m.group(1).strip()
  17. # Missing the final answer line
  18. if final_score is None:
  19. return None
  20. else:
  21. return float(final_score) > 0
  22. def main(args):
  23. default_tabulate(args, scorer=scorer)
  24. if __name__ == "__main__" and __package__ is None:
  25. main(sys.argv)