docs: update api references

metaopt · Nov 19, 2022 · 15d00d2 · 15d00d2
1 parent c9c17a9
commit 15d00d2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/docs/source/api/api.rst b/docs/source/api/api.rst
@@ -187,31 +187,31 @@ Optimizer Hooks
     register_hook
     zero_nan_hook
     nan_to_num_hook
-    nan_to_num
 
 Hook
 ~~~~
 
 .. autofunction:: register_hook
 .. autofunction:: zero_nan_hook
 .. autofunction:: nan_to_num_hook
-.. autofunction:: nan_to_num
 
 ------
 
 Gradient Transformation
 =======================
 
-.. currentmodule:: torchopt.clip
+.. currentmodule:: torchopt
 
 .. autosummary::
 
     clip_grad_norm
+    nan_to_num
 
 Transforms
 ~~~~~~~~~~
 
 .. autofunction:: clip_grad_norm
+.. autofunction:: nan_to_num
 
 Optimizer Schedules
 ===================