Docstring and type hints for the prediction module

[codecov-commenter]

Codecov Report

All modified and coverable lines are covered by tests :white_check_mark:

Project coverage is 23.04%. Comparing base (08d8c97) to head (2b5ba65).

:exclamation: Your organization needs to install the Codecov GitHub app to enable full functionality.

Additional details and impacted files

```diff @@ Coverage Diff @@ ## master #666 +/- ## ======================================= Coverage 23.03% 23.04% ======================================= Files 161 161 Lines 27761 27763 +2 ======================================= + Hits 6396 6398 +2 Misses 21365 21365 ```

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.