* [Fix]: Fix data batch type in base hook

* [Fix]: Fix the type hint bug in checkpoint, optimizer, param scheduler hooks

Co-authored-by: Your <you@example.com>

* [Feature]: Add param scheduler hook

* [Fix]: update docstring and add assert_call to UT