Import and return the requested module.

def load_module(module):
    """
    Import and return the requested module.
    """
    try:
        m = importlib.import_module(module)
        return m
    except ModuleNotFoundError:  # pragma: no cover
        raise TImportError("Module `%s` is not installed. It is required for "
                           "running this function." % module)