Help me catch defining actions that return context without being wrapped in assign

[rostero1]

I'm constantly forgetting to wrap my actions with assign. It'd be nice if I could get a warning when this happens. Or if even if my return includes keys present in context (which seems a little hacky).