Function dgl_warning

python/dgl/base.py:51–53

DGL warning wrapper that defaults to ``DGLWarning`` instead of ``UserWarning`` category.

(message, category=DGLWarning, stacklevel=2)

def dgl_warning(message, category=DGLWarning, stacklevel=2):
    """DGL warning wrapper that defaults to ``DGLWarning`` instead of ``UserWarning`` category."""
    return warnings.warn(message, category=category, stacklevel=stacklevel)


warnings.formatwarning = dgl_warning_format