Fix incorrect function reference in documentation.

PiperOrigin-RevId: 477970031
google-deepmind · Sep 30, 2022 · 96e513a · 96e513a
1 parent 560510c
commit 96e513a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/haiku/_src/base.py b/haiku/_src/base.py
@@ -779,7 +779,7 @@ def name(self):
 def custom_setter(setter: Setter) -> contextlib.AbstractContextManager:
   """Registers a custom state setter.
 
-  When state is set using :func:`get_parameter` we always run all custom setters
+  When state is set using :func:`set_state` we always run all custom setters
   before saving the value.
 
   >>> def zero_during_init(next_setter, value, context):