From 0da8a2bd998e33128b66cbcaa6489ad72c1aed7e Mon Sep 17 00:00:00 2001 From: merwanehamadi Date: Sun, 13 Aug 2023 10:27:54 -0700 Subject: [PATCH] Fix agent protocol test (#301) Signed-off-by: Merwane Hamadi --- agbenchmark/utils/challenge.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/agbenchmark/utils/challenge.py b/agbenchmark/utils/challenge.py index 9a08cb0a..19593b72 100644 --- a/agbenchmark/utils/challenge.py +++ b/agbenchmark/utils/challenge.py @@ -47,14 +47,15 @@ class Challenge(ABC): return self.data.dependencies async def setup_challenge(self, config: Dict[str, Any], cutoff: int) -> None: - if not self.task: - return - from agbenchmark.agent_interface import copy_artifacts_into_workspace, run_agent copy_artifacts_into_workspace( config["workspace"], "artifacts_in", self.ARTIFACTS_LOCATION ) + if not self.task: + return + + print( f"\033[1;35m============Starting {self.data.name} challenge============\033[0m"