fftconvolve usage fixed

that had broke due to the move to a different file

Author: sambit-giri

src/tools21cm/beam_convolve.py

@@ -3,7 +3,7 @@
 from scipy import signal
 from .helper_functions import print_msg
 from .smoothing import gauss_kernel, get_beam_w
-from .helper_functions import fftconvolve
+from .fft_functions import fftconvolve
 
 def beam_convolve(input_array, z, fov_mpc, beam_w = None, max_baseline = None, \
 				beamshape='gaussian'):

src/tools21cm/plotting.py

@@ -191,7 +191,9 @@ def plot_triangle(samples_dict, weights_dict=None,
             )
         # Add legend
         handles = [mlines.Line2D([], [], color=(colors[n] if isinstance(colors, dict) else colors), label=n) for n in names]
-        fig.legend(handles=handles, loc='upper right', bbox_to_anchor=(0.95, 0.95))
+        bbox_to_anchor = kwargs.get('bbox_to_anchor', (0.95, 0.95))
+        legend_fontsize = kwargs.get('legend_fontsize', 14)
+        fig.legend(handles=handles, loc='upper right', bbox_to_anchor=bbox_to_anchor, fontsize=legend_fontsize)
         return fig
 
     # --- 2. GETDIST BACKEND ---

src/tools21cm/smoothing.py

@@ -13,7 +13,8 @@
 from numpy.fft import rfftn, irfftn
 from math import ceil, floor
 from numpy import array, asarray, roll
-from .helper_functions import fftconvolve, find_idx
+from .helper_functions import find_idx
+from .fft_functions import fftconvolve
 from tqdm import tqdm
 import astropy.units as u