Valid linting when using register_buffer/register_parameter?

Hi!

I am using register_buffer and register_parameter to initialise a parameter for a pytorch model. However, when doing so, the attribute that I have set via either function does not get picked up as a class attribute by the python linter. Has anyone managed to find a way to address this, please?
Thanks