touchups

ratt-ru · Oct 30, 2024 · d10c748 · d10c748
1 parent 16d888c
commit d10c748
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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):