diff --git a/run.py b/run.py index cce45bc..75148e3 100644 --- a/run.py +++ b/run.py @@ -218,7 +218,7 @@ def test(args: argparse.Namespace, test_all_meta: dict) -> None: f.write("\n") env.close() - logger.info(f"Average score: {sum(scores) / len(scores)}") + logger.info(f"Average score: {sum(scores) / len(scores) if scores else 0}") def get_unfinished( diff --git a/run_autoglm.py b/run_autoglm.py index 8b161c7..5776886 100644 --- a/run_autoglm.py +++ b/run_autoglm.py @@ -457,7 +457,7 @@ def test(args: argparse.Namespace, test_all_meta: dict) -> None: f.write("\n") env.close() - logger.info(f"Average score: {sum(scores) / len(scores)}") + logger.info(f"Average score: {sum(scores) / len(scores) if scores else 0}") def get_unfinished(action_space, use_model, observation_type, result_dir, total_file_json): diff --git a/run_autoglm_v.py b/run_autoglm_v.py index 31794ea..0f95af0 100644 --- a/run_autoglm_v.py +++ b/run_autoglm_v.py @@ -485,7 +485,7 @@ def test(args: argparse.Namespace, test_all_meta: dict) -> None: f.write("\n") env.close() - logger.info(f"Average score: {sum(scores) / len(scores)}") + logger.info(f"Average score: {sum(scores) / len(scores) if scores else 0}") def get_unfinished(action_space, use_model, observation_type, result_dir, total_file_json):