Allows for beam simulation while having maximum one dimension larger than 1 element

Author: AlanLoh

nenupy/__init__.py

@@ -5,7 +5,7 @@
 __copyright__ = "Copyright 2023, nenupy"
 __credits__ = ["Alan Loh"]
 __license__ = "MIT"
-__version__ = "2.6.1"
+__version__ = "2.6.2"
 __maintainer__ = "Alan Loh"
 __email__ = "alan.loh@obspm.fr"
 

nenupy/instru/nenufar.py

@@ -145,7 +145,11 @@ def __getitem__(self, sky: Sky) -> np.ndarray:
             ) for gain_i in gain
         ])
 
-        if gain.ndim == 2:
+        if gain.ndim == 1:
+            # If only one or less dimension is larger than 1 element, get_interp_val returns a 1D array
+            # It's then esay to just reshape like the original array since a single dimension is affected at best
+            gain = gain.reshape(sky.value.shape)
+        elif gain.ndim == 2:
             # The time dimension is not yet included
             gain = gain.reshape((1,) + gain.shape)
         # Return something shaped as (time, freq, coord)
@@ -915,6 +919,7 @@ def _antenna_gain(self, sky: Sky, pointing: Pointing):
             ),
             dtype=np.float64
         )
+        log.debug(f"Antenna gain shape: {gain.shape}.")
         for i, pol in enumerate(sky.polarization):
             if not isinstance(pol, Polarization):
                 log.warning(