Update function arguments typing

[amorehead]

What does this PR do?

Updates function arguments typing.

Before submitting

• [x] Did you make sure title is self-explanatory and the description concisely explains the PR?
• [x] Did you make sure your PR does only one thing, instead of bundling different changes together?
• [x] Did you list all the breaking changes introduced by this pull request?
• [ ] Did you test your PR locally with pytest command?
• [ ] Did you run pre-commit hooks with pre-commit run -a command?

Did you have fun?

Make sure you had fun coding 🙃

[codecov-commenter]

Codecov Report

Patch coverage: 100.00% and no project coverage change.

Comparison is base (2654bad) 84.03% compared to head (0f18956) 84.03%.

Additional details and impacted files

```diff @@ Coverage Diff @@ ## main #603 +/- ## ======================================= Coverage 84.03% 84.03% ======================================= Files 11 11 Lines 357 357 ======================================= Hits 300 300 Misses 57 57 ``` | [Files Changed](https://app.codecov.io/gh/ashleve/lightning-hydra-template/pull/603?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=None) | Coverage Δ | | |---|---|---| | [src/utils/utils.py](https://app.codecov.io/gh/ashleve/lightning-hydra-template/pull/603?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=None#diff-c3JjL3V0aWxzL3V0aWxzLnB5) | `72.09% <100.00%> (ø)` | |

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