fix(mpiarray): fix doctest warning

radiocosmology · Sep 26, 2024 · fcd4307 · fcd4307
1 parent e02b8c2
commit fcd4307
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/caput/mpiarray.py b/caput/mpiarray.py
@@ -19,7 +19,7 @@
 
     # Load in data into parallel array
     for lfi, fi in darr1.enumerate(axis=0):
-        darr1[lfi] = load_freq_data(gfi)
+        darr1.local_array[lfi] = load_freq_data(gfi)
 
     # Perform m-transform (i.e. FFT)
     darr2 = MPIArray.wrap(np.fft.fft(darr1, axis=1), axis=0)