Merge pull request #571 from VChristiaens/master

Now closing HDU in open_fits function, PEP8 formatting and upgraded version
vortex-exoplanet · Jan 9, 2023 · 9948e7f · 9948e7f
2 parents 0ab7a89 + 88b0edc
commit 9948e7f
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 12 deletions.
diff --git a/paper.pdf b/paper.pdf
diff --git a/vip_hci/__init__.py b/vip_hci/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "1.3.5"
+__version__ = "1.3.6"
 
 
 from . import preproc

diff --git a/vip_hci/fits/fits.py b/vip_hci/fits/fits.py
@@ -70,7 +70,7 @@ def open_fits(fitsfilename, n=0, header=False, ignore_missing_end=False,
     else:
         data = hdulist[n].data
         data = np.array(data, dtype=precision)
-
+        hdulist.close()
         if header:
             header = hdulist[n].header
             if verbose:

diff --git a/vip_hci/preproc/badpixremoval.py b/vip_hci/preproc/badpixremoval.py
@@ -778,10 +778,10 @@ def bp_removal_2d(array_corr, cy, cx, fwhm, sig, protect_mask, bpm_mask_ori,
         if bpm_mask is None or not correct_only:
             if (cy is None or cx is None) and protect_mask:
                 cy, cx = frame_center(array)
-            array_corr, bpix_map_cumul = bp_removal_2d(array_corr, cy, cx, fwhm, sig,
-                                                    protect_mask, bpm_mask,
-                                                    min_thr, half_res_y, mad,
-                                                    verbose)
+            array_corr, bpix_map_cumul = bp_removal_2d(array_corr, cy, cx, fwhm,
+                                                       sig, protect_mask,
+                                                       bpm_mask, min_thr,
+                                                       half_res_y, mad, verbose)
         else:
             fwhm_round = int(round(fwhm))
             fwhm_round = fwhm_round+1-(fwhm_round % 2)  # make it odd
@@ -807,12 +807,14 @@ def bp_removal_2d(array_corr, cy, cx, fwhm, sig, protect_mask, bpm_mask_ori,
             for i in range(n_z):
                 if verbose:
                     print('************Frame # ', i, ' *************')
-                array_corr[i], bpix_map_cumul[i] = bp_removal_2d(array_corr[i], cy[i],
-                                                              cx[i], fwhm[i],
-                                                              sig, protect_mask,
-                                                              bpm_mask, min_thr,
-                                                              half_res_y, mad,
-                                                              verbose)
+                array_corr[i], bpix_map_cumul[i] = bp_removal_2d(array_corr[i],
+                                                                 cy[i], cx[i],
+                                                                 fwhm[i], sig,
+                                                                 protect_mask,
+                                                                 bpm_mask,
+                                                                 min_thr,
+                                                                 half_res_y, mad,
+                                                                 verbose)
         else:
             if isinstance(fwhm, (float, int)):
                 fwhm_round = int(round(fwhm))