train_dataset = Dataset.from_dict(
    {
        "anchor": [
            "reset a locked account",
            "reset a user password",
            "enable two-factor authentication",
            "rotate an API token",
            "restore a deleted project",
            "invite a new support agent",
            "export audit logs",
            "change the billing contact",
        ],
        "positive": [
            "Unlock the account from the admin users page.",
            "Open the user profile and send a password reset email.",
            "Open account security and enroll an authenticator app.",
            "Revoke the old API token and create a replacement token.",
            "Open deleted projects and restore the selected project.",
            "Open team settings and send an invitation email.",
            "Open compliance reports and export the audit log CSV.",
            "Open billing settings and update the primary contact.",
        ],
    }
)