Skip to content

Commit b232d71

Browse files
chore: refactor from dn.score (#17)
1 parent ab56e10 commit b232d71

File tree

3 files changed

+173
-9
lines changed

3 files changed

+173
-9
lines changed

airtbench/main.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -334,15 +334,14 @@ async def check_for_flags(content: str) -> bool:
334334
logger.success(f"|- Got the flag for {challenge.id}!")
335335
dn.log_metric("found_flag", 1)
336336

337-
flag_score = dn.Score(
338-
name="flag_found",
339-
value=1.0,
337+
dn.log_metric(
338+
"flag_found",
339+
1.0,
340340
attributes={
341341
"challenge_id": challenge.id,
342342
"flag": match[:10] + "...",
343343
},
344344
)
345-
dn.log_score(flag_score)
346345
return True
347346
return False
348347

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ dependencies = [
77
"backoff>=2.2.1",
88
"cyclopts>=3.12.0",
99
"docker>=7.1.0",
10-
"dreadnode>=1.0.0rc1",
10+
"dreadnode==1.0.6",
1111
"ipykernel>=6.29.5",
1212
"mypy>=1.15.0",
1313
"pythonnet>=3.0.5",

0 commit comments

Comments
 (0)