diff --git a/imperative/python/megengine/functional/metric.py b/imperative/python/megengine/functional/metric.py
index 77dd63af15cfe3794e95078409a3f11d15b9c17f..ee010a11f71ab6b8695f02dff7f982dad0b8c6ca 100644
--- a/imperative/python/megengine/functional/metric.py
+++ b/imperative/python/megengine/functional/metric.py
@@ -15,6 +15,10 @@ from .elemwise import abs, maximum, minimum
 from .math import topk as _topk
 from .tensor import broadcast_to, transpose
 
+__all__ = [
+    "topk_accuracy",
+]
+
 
 def topk_accuracy(
     logits: Tensor, target: Tensor, topk: Union[int, Iterable[int]] = 1
diff --git a/imperative/python/megengine/functional/nn.py b/imperative/python/megengine/functional/nn.py
index 305c30f7e9a2d1c8678db248cb96428c2139a214..568df43fb38bc62d7ae0349d80e11ce3ef35d24c 100644
--- a/imperative/python/megengine/functional/nn.py
+++ b/imperative/python/megengine/functional/nn.py
@@ -1660,3 +1660,4 @@ warp_perspective = deprecated_func(
 )
 from .quantized import conv_bias_activation  # isort:skip
 from .loss import *  # isort:skip
+from .metric import *  # isort:skip