Skip to content

FEAT: add harm categories to AdvBench Dataset - #732

Merged
Roman Lutz (romanlutz) merged 13 commits into
microsoft:mainfrom
paulinek13:adv_bench_dataset
Mar 11, 2025
Merged

FEAT: add harm categories to AdvBench Dataset#732
Roman Lutz (romanlutz) merged 13 commits into
microsoft:mainfrom
paulinek13:adv_bench_dataset

Conversation

@paulinek13

@paulinek13 Paulina Kalicka (paulinek13) commented Feb 22, 2025

Copy link
Copy Markdown
Contributor

Description

This PR aims to resolve #730 by adding harm categories to the AdvBench dataset and enabling filtering support based on those categories.

Tests and Documentation

✔️ added tests
✔️ updated the docstring

Comment thread pyrit/datasets/fetch_example_datasets.py Outdated
Comment thread pyrit/datasets/harm_categories/adv_bench_dataset.json Outdated
@paulinek13 Paulina Kalicka (paulinek13) changed the title [DRAFT] add harm categories to AdvBench Dataset FEAT: add harm categories to AdvBench Dataset Mar 10, 2025
@paulinek13
Paulina Kalicka (paulinek13) marked this pull request as ready for review March 10, 2025 16:14
Comment thread pyrit/datasets/fetch_example_datasets.py
Comment thread pyrit/datasets/fetch_example_datasets.py

@romanlutz Roman Lutz (romanlutz) left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Awesome! LGTM

@romanlutz
Roman Lutz (romanlutz) merged commit 7138510 into microsoft:main Mar 11, 2025
@paulinek13
Paulina Kalicka (paulinek13) deleted the adv_bench_dataset branch March 13, 2025 10:47
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

BUG harm categories for AdvBench Dataset aren't added yet

2 participants