blabla

dmlc · Feb 9, 2024 · af7c3a6 · af7c3a6
1 parent 7fc6d3a
commit af7c3a6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/dgl/base.py b/python/dgl/base.py
@@ -53,7 +53,7 @@ def dgl_warning(message, category=DGLWarning, stacklevel=2):
     """DGL warning wrapper that defaults to ``DGLWarning`` instead of ``UserWarning`` category."""
     # Skip warning in unit test.
     if "unittest" in sys.modules.keys():
-        return
+        return None
     return warnings.warn(message, category=category, stacklevel=stacklevel)