DataParallel-safe method call in model

If the model is wrapped with DataParallel class or something, i have to call the method like “model.module.method”
But it is easy to forget that the model is wrapped with some classes.

Is there any wrapper-safe calls to prevent these errors?