How do I set the state_handlers for a task if I am using a custom class that inherits from Task and I have my own init and run method ?

It should work the same way. Rather than attaching the state handler on a task decorator, you can set it when initializing the task

Call the super init like this