CI: Fix mypy failures
- [x] closes #44143

> Thanks @alimcmaster1 merge on green

ty - will do! Thanks for the prompt review! Shouldn't the type (of the input and output) be changed to np.ndarray as the dtype might change? LGTM /azp run <samp>
Azure Pipelines successfully started running 1 pipeline(s).<br>

</samp>