update code with type hints

[EnggQasim]

Description of the change

Update type hints code

Motivation

python type hints help to developer for coding time

Type of change

[google-cla]

Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA).

View this failed invocation of the CLA check for more information.

For the most up to date status, view the checks section at the bottom of the pull request.

[review-notebook-app]

Check out this pull request on 

See visual diffs & provide feedback on Jupyter Notebooks.

---

Powered by ReviewNB

[github-actions]

Marking this pull request as stale since it has been open for 14 days with no activity. This PR will be closed if no further activity occurs.

[MarkDaoust]

@EnggQasim we can't move forward with this if you don't sign the CLA.

[github-actions]

Marking this pull request as stale since it has been open for 14 days with no activity. This PR will be closed if no further activity occurs.

[github-actions]

This pull request was closed because it has been inactive for 28 days. Please open a new pull request if you need further assistance. Thanks!