diff --git a/africanus/experimental/rime/fused/transformers/parangle.py b/africanus/experimental/rime/fused/transformers/parangle.py index a4a9bf61..15b0d119 100644 --- a/africanus/experimental/rime/fused/transformers/parangle.py +++ b/africanus/experimental/rime/fused/transformers/parangle.py @@ -66,10 +66,10 @@ def parangles(init_state, antenna_position, phase_dir, receptor_angle=None): raise ValueError("receptor_angle.ndim != 2") if receptor_angle.shape[1] != 2: - raise ValueError("Only 2 receptor angles " "currently supported") + raise ValueError("Only 2 receptor angles currently supported") # Select out the feeds we're interested in - receptor_angle = receptor_angle[ufeed, :] + receptor_angle = receptor_angle[init_state.ufeed, :] for t in range(ntime): for f in range(nfeed):