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