Address Aarynn's comments part 4

spacetelescope · Jul 30, 2024 · 0d0e04d · 0d0e04d
1 parent 40d654d
commit 0d0e04d
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/spaceKLIP/analysistools.py b/spaceKLIP/analysistools.py
@@ -390,13 +390,11 @@ def raw_contrast(self,
                     columns = [seps[0]]
                     names = ['separation']
                     for i, klmode in enumerate(klmodes):
-                        if mask is None:
-                            columns.append(cons[i])
-                        else:
-                            columns.append(cons_mask[i])
+                        columns.append(cons[i])
                         names.append(f'contrast, N_kl={klmode}')
-                        columns.append(cons_mask[i])
-                        names.append(f'contrast+mask, N_kl={klmode}')
+                        if mask is not None:
+                            columns.append(cons_mask[i])
+                            names.append(f'contrast+mask, N_kl={klmode}')
                     results_table = Table(columns,
                                           names=names)
                     results_table['separation'].unit = u.arcsec