def fetch_pku_safe_rlhf_dataset()

in pyrit/datasets/pku_safe_rlhf_dataset.py [0:0]

• 27 lines of code
• 1 McCabe index (conditional complexity)

def fetch_pku_safe_rlhf_dataset(
    include_safe_prompts: bool = True,
    filter_harm_categories: Optional[
        List[
            Literal[
                "Animal Abuse",
                "Copyright Issues",
                "Cybercrime",
                "Discriminatory Behavior",
                "Disrupting Public Order",
                "Drugs",
                "Economic Crime",
                "Endangering National Security",
                "Endangering Public Health",
                "Environmental Damage",
                "Human Trafficking",
                "Insulting Behavior",
                "Mental Manipulation",
                "Physical Harm",
                "Privacy Violation",
                "Psychological Harm",
                "Sexual Content",
                "Violence",
                "White-Collar Crime",
            ]
        ]
    ] = None,