from astroid import MANAGER
from astroid import nodes

def register(linter):
    MANAGER.register_transform(nodes.Module, transform)

def transform(module):
    #print("processing",module.name)
    if module.name.startswith('numpy'):
        if module.name == 'numpy':
            import numpy
        elif module.name == 'numpy.random':
            import numpy.random
            from numpy.random import seed, get_state, set_state