Callbacks Typing should be updated

[tshu-w]

as the title suggests https://github.com/PyTorchLightning/pytorch-lightning/blob/master/pytorch_lightning/callbacks/base.py

[awaelchli]

Feel free to open a pull request if you see missing or wrong type hints. The help would be very much appreciated, thanks :)