fix mistakes in type hinting
Builds on !96 (merged).
Few type hints are not complete in the main branch; This MR fixes them.
Edited by Ashwin Kumar Karnad
Builds on !96 (merged).
Few type hints are not complete in the main branch; This MR fixes them.