def METHOD_NAME(self, original):
    @functools.wraps(original)
    def wrapper(name, *args, **kwargs):
        self._record_imported_module(name)
        return original(name, *args, **kwargs)
    return wrapper