I think the latest version of

flytekit

might have an issue with custom exceptions, I’ve created an issue