Delete the useless ids; Fix bugs in run.py

This commit is contained in:
Timothyxxx
2024-03-16 14:12:54 +08:00
parent 04fcb9d971
commit 5a062c423f
2 changed files with 1 additions and 4 deletions

3
run.py
View File

@@ -79,7 +79,7 @@ def config() -> argparse.Namespace:
# agent config
parser.add_argument("--max_trajectory_length", type=int, default=3)
parser.add_argument("--test_config_base_dir", type=str, default="evaluation_examples")
parser.add_argument("--example_time_limit", type=int, default=600)
parser.add_argument("--example_time_limit", type=int, default=1200)
# lm config
parser.add_argument("--model", type=str, default="gpt-4-vision-preview")
@@ -211,5 +211,4 @@ if __name__ == '__main__':
for domain in test_file_list:
left_info += f"{domain}: {len(test_file_list[domain])}\n"
logger.info(f"Left tasks:\n{left_info}")
os.environ["OPENAI_API_KEY"] = "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"
test(args, test_all_meta)