NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-19 12:57:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update refusalbench_environment.py

Browse source
This commit is contained in:
VolodymyrBg 2026-01-29 12:43:15 +02:00 • committed by GitHub
parent 94f29eac18
commit f285bbd417
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
environments/eval_environments/refusalbench_environment.py
View file

@ -447,7 +447,7 @@ class RefusalBenchEnv(BaseEnv):
total_train_items += 1
category = item.get(
"overarching_category", "Unknown"
) # The category needs to be called overarching_category. Otherwise it wont work
) # The category needs to be called overarching_category. Otherwise it won't work
train_category_counts[category] = (
train_category_counts.get(category, 0) + 1
)