Don't write fisher samples

Author: andreicuceu

lya_2pt/interface.py

@@ -300,12 +300,12 @@ def run(self, healpix_ids=None, mpi_size=1, mpi_rank=0):
                         results = list(pool.imap_unordered(compute_xi_and_fisher, local_hp_ids))
 
                 for hp_id, res in results:
-                    self.optimal_xi_output[hp_id] = res
+                    self.optimal_xi_output[hp_id] = res[0]
                     self.xi_est += res[0]
                     self.fisher_est += res[1]
             else:
                 for healpix_id in healpix_ids:
-                    self.optimal_xi_output[healpix_id] = compute_xi_and_fisher(healpix_id)[1]
+                    self.optimal_xi_output[healpix_id] = compute_xi_and_fisher(healpix_id)[1][0]
 
                     self.xi_est += self.optimal_xi_output[healpix_id][0]
                     self.fisher_est += self.optimal_xi_output[healpix_id][1]

lya_2pt/output.py

@@ -1,4 +1,5 @@
 import fitsio
+import numpy as np
 
 from lya_2pt.utils import check_dir, parse_config, find_path
 
@@ -225,14 +226,15 @@ def write_optimal_cf(
         filename = self.healpix_dir / f"opt-corr-samples{proc_string}.fits"
         output_fits = fitsio.FITS(filename, 'rw', clobber=True)
 
-        for healpix_id, result in output.items():
-            output_fits.write(
-                [result[0], result[1]],
-                names=[correlation_name, "FISHER_MATRIX"],
-                comment=['unnormalized correlation', 'Fisher matrix'],
-                header=header,
-                extname=str(healpix_id)
-            )
+        correlation_samples = np.array([result for result in output.values()])
+
+        output_fits.write(
+            [correlation_samples],
+            names=[correlation_name],
+            comment=['unnormalized correlation'],
+            header=header,
+            extname='SAMPLES'
+        )
 
         output_fits.close()