Diagonals functions currently support dask arrays, but we were still passing them numpy ones.
This PR fixes this by using dask arrays for the data to average
Diagonals functions currently support dask arrays, but we were still passing them numpy ones.
This PR fixes this by using dask arrays for the data to average