pyrit.datasets.fetch_equitymedqa_dataset_unique_values

pyrit.datasets.fetch_equitymedqa_dataset_unique_values#

fetch_equitymedqa_dataset_unique_values(subset_name: Literal['all'] | str | Sequence[str] = 'all') SeedDataset[source]#

Fetch the EquityMedQA dataset from Hugging Face and return a SeedDataset.

Parameters:

subset_name (str | list) – The name(s) of the subset to fetch. Defaults to “all” which returns all values.

Returns:

A SeedDataset containing the unique prompts from the dataset.

Return type:

SeedDataset

Raises:

ValueError – If any of the specified subset names are invalid.

Notes: For more info

Paper: https://arxiv.org/abs/2403.12025