diff --git a/declearn/optimizer/schedulers/_api.py b/declearn/optimizer/schedulers/_api.py index 5a3a7e5bcf9530f8bb74b035aabc85ec7ac0dc99..d14cc27dabf45dde6cd3575c43470d52c3c7f4d2 100644 --- a/declearn/optimizer/schedulers/_api.py +++ b/declearn/optimizer/schedulers/_api.py @@ -99,10 +99,10 @@ class Scheduler(metaclass=abc.ABCMeta): Inheritance ----------- - When a subclass inheriting from `OptiModule` is declared, it is - automatically registered under the "OptiModule" group using its + When a subclass inheriting from `Scheduler` is declared, it is + automatically registered under the "Scheduler" group using its class-attribute `name`. This can be prevented by adding `register=False` - to the inheritance specs (e.g. `class MyCls(OptiModule, register=False)`). + to the inheritance specs (e.g. `class MyCls(Scheduler, register=False)`). See `declearn.utils.register_type` for details on types registration. """