The current way of patching swaps the kafka classes with AutoInstrumented* versions, and then wrap their methods.
That has the drawback of not being able to patch (in tests at least) if these classes have already been imported. I guess we can patch the kafka classes directly as the other instrumentations usually do.