("three", "source", "time", "ant", "chan"),
coeffs, ("ant", "chan") + corr_dims + ("poly", ),
noll_index,
("ant", "chan") + corr_dims + ("poly", ),
dtype=coeffs.dtype)
try:
phase_delay.__doc__ = PHASE_DELAY_DOCS.substitute(
array_type=":class:`dask.array.Array`")
except AttributeError:
pass
try:
parallactic_angles.__doc__ = mod_docs(
np_parangles.__doc__,
[(":class:`numpy.ndarray`", ":class:`dask.array.Array`")])
except AttributeError:
pass
try:
feed_rotation.__doc__ = FEED_ROTATION_DOCS.substitute(
array_type=":class:`numpy.ndarray`")
except AttributeError:
pass
try:
transform_sources.__doc__ = mod_docs(
np_transform_sources.__doc__,
[(":class:`numpy.ndarray`", ":class:`dask.array.Array`")])
except AttributeError:
assert grid_flat_corrs == weight_flat_corrs
assert uvw.shape[0] == weights.shape[0]
assert weights.shape[1] == ref_wave.shape[0]
# Creation correlation dimension strings for each correlation
corrs = tuple('corr-%d' % i for i in range(len(grid.shape[2:])))
return da.core.blockwise(np_degrid_fn, ("row", "chan") + corrs,
grid, ("ny", "nx") + corrs,
uvw, ("row", "(u,v,w)"),
weights, ("row", "chan") + corrs,
ref_wave, ("chan",),
concatenate=True,
convolution_filter=convolution_filter,
cell_size=cell_size,
dtype=np.complex64)
grid.__doc__ = mod_docs(np_grid_fn.__doc__,
[(":class:`numpy.ndarray`",
":class:`dask.array.Array`"),
("np.ones_like", "da.ones_like"),
("np.zeros_like", "da.zeros_like")])
degrid.__doc__ = mod_docs(np_degrid_fn.__doc__,
[(":class:`numpy.ndarray`",
":class:`dask.array.Array`"),
("np.ones_like", "da.ones_like"),
("np.zeros_like", "da.zeros_like")])