pyrit.datasets.fetch_babelscape_alert_dataset#
- fetch_babelscape_alert_dataset(category: Literal['alert', 'alert_adversarial'] | None = 'alert_adversarial') SeedPromptDataset[source]#
Fetch the Babelscape/ALERT dataset and create a SeedPromptDataset.
- Parameters:
category (Optional[str]) – The dataset category, “alert” or “alert_adversarial”. If None, both categories will be loaded. Defaults to “alert_adversarial”.
- Returns:
A SeedPromptDataset containing the examples.
- Return type: