datasets_loader.py

from datasets import load_dataset

def load_redteamer_mistral():
    return load_dataset("romaingrx/red-teamer-mistral-", split="train")

def load_multi_class_redteaming():
    return load_dataset("SummerSigh/Muti-Class-Redteaming", split="train")