def METHOD_NAME(event):
    if isinstance(event.input, dict) and "__extra" in event.input:
        return event.input["__extra"]["class"]
    workflow_name = event.workflow_type["name"]
    return workflow_name