dev(Accuracy): unifiy docstrings

open-mmlab · Sep 29, 2022 · e48d94a · e48d94a
1 parent b840f72
commit e48d94a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/mmeval/classification/accuracy.py b/mmeval/classification/accuracy.py
@@ -27,16 +27,16 @@ def _numpy_topk(inputs: np.ndarray,
     This implementation returns the values and indices of the k largest
     elements along a given axis.
 
-    Note:
-        If PyTorch is available, the ``_torch_topk`` would be used.
-
     Args:
         inputs (nump.ndarray): The input numpy array.
         k (int): The k in `top-k`.
         axis (int, optional): The axis to sort along.
 
     Returns:
         tuple: The values and indices of the k largest elements.
+
+    Note:
+        If PyTorch is available, the ``_torch_topk`` would be used.
     """
     if torch is not None:
         values, indices = _torch_topk(torch.from_numpy(inputs), k, dim=axis)