task_categories:conversational size_categories:1K<n<10K language:en license:mit art region:us
---
license: mit
task_categories:
- conversational
language:
- en
tags:
- art
size_categories:
- 1K<n<10K
---

**NOTE** This dataset is under development.

To load the dataset and prepare it for DPO training.

```python
from datasets import load_dataset
ds = load_dataset("plaguss/the_office_ds_dwight_qa")

def return_prompt_and_responses(samples) -> dict[str, str]:
    return {
        "prompt": "Question: " + samples["prompt"] + "\n\nAnswer: ",
        "chosen": samples[samples["label"]],
        "rejected": samples["response_2" if samples["label"] == "response_1" else "response_1"],
    }

rm_columns = ["response_1", "response_2", "is_dwight"]

ds = ds.map(
    return_prompt_and_responses,
    remove_columns=rm_columns
)
```