exovetter 0.0.9__tar.gz → 0.0.11__tar.gz

{exovetter-0.0.9 → exovetter-0.0.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: exovetter
-Version: 0.0.9
+Version: 0.0.11
 Summary: Exoplanet vetting package
 Home-page: https://github.com/spacetelescope/exovetter
 Author: Susan Mullally et al.

{exovetter-0.0.9 → exovetter-0.0.11}/exovetter/centroid/centroid.py

@@ -4,6 +4,7 @@ import exovetter.centroid.disp as disp
 import exovetter.utils as utils
 import matplotlib.pyplot as plt
 import numpy as np
+import warnings
 
 
 def compute_diff_image_centroids(
@@ -12,7 +13,7 @@ def compute_diff_image_centroids(
         period_days, 
         epoch, 
         duration_days,
-        remove_transits, 
+        remove_transits=None, 
         max_oot_shift_pix=1.5,
         starloc_pix = None,
         plot=False
@@ -71,6 +72,9 @@ def compute_diff_image_centroids(
     OOT from In-transit.
     """
 
+    if remove_transits is None:
+            remove_transits = []
+        
     isnan = np.isnan(time)
     time = time[~isnan]
     cube = cube[~isnan]
@@ -135,7 +139,18 @@ def measure_centroid_shift(centroids, kept_transits, plot=False):
     dcol = centroids[:, 4] - centroids[:, 0]
     drow = centroids[:, 5] - centroids[:, 1]
 
+    if centroids.size == 0:
+            raise RuntimeError("No difference-image centroids could be computed for this target.")
+            
     flags = centroids[:, -1].astype(bool)
+    N_use = np.sum(~flags) # Check if there are enough TPFs left to actually centroid
+        
+    if N_use < 2:
+            raise RuntimeError("TPF data quality flagged too many difference-image centroids "
+                               "(<2 usable transits); centroid significance cannot be computed.")
+    elif N_use == 2:
+            warnings.warn("Only 2 usable difference-image centroids available; "
+                          "centroid significance is not statistically reliable.", RuntimeWarning, stacklevel=2)
 
     offset_pix, signif = covar.compute_offset_and_signif(
         dcol[~flags], drow[~flags])
@@ -179,7 +194,18 @@ def measure_centroid_shift_cat(centroids, kept_transits, starloc_pix, plot=False
     dcol = centroids[:, 4] - starloc_pix[0]
     drow = centroids[:, 5] - starloc_pix[1]
 
+    if centroids.size == 0:
+            raise RuntimeError("No difference-image centroids could be computed for this target.")
+    
     flags = centroids[:, -1].astype(bool)
+    N_use = np.sum(~flags)
+
+    if N_use < 2:
+            raise RuntimeError("TPF data quality flagged too many difference-image centroids "
+                               "(<2 usable transits); centroid significance cannot be computed.")
+    elif N_use == 2:
+            warnings.warn("Only 2 usable difference-image centroids available; "
+                          "centroid significance is not statistically reliable.", RuntimeWarning, stacklevel=2)
 
     offset_pix, signif = covar.compute_offset_and_signif(
         dcol[~flags], drow[~flags])

{exovetter-0.0.9 → exovetter-0.0.11}/exovetter/version.py

@@ -5,4 +5,4 @@ try:
     from setuptools_scm import get_version
     version = get_version(root='..', relative_to=__file__)
 except Exception:
-    version = '0.0.9'
+    version = '0.0.11'

{exovetter-0.0.9 → exovetter-0.0.11}/exovetter/vetters.py

@@ -710,7 +710,8 @@ class VizTransits(BaseVetter):
             Zoom in on the transit
 
         smooth : type
-            description
+            Number of points across 3 in-transit durations for a 1DBoxkernel. 
+            Default is 10. None will turn off smoothing.
 
         transit_plot : bool
             Whether or not to show the transit plot
@@ -1109,4 +1110,4 @@ def run_all(tce, lc, vetters=[VizTransits(), ModShift(), Lpp(), OddEven(), Trans
     # if verbose:
     #     print('Execution time:', (py_time.time() - run_start), 's')
 
-    # return results_df
+    # return results_df

{exovetter-0.0.9 → exovetter-0.0.11}/exovetter/viz_transits.py

@@ -177,7 +177,7 @@ def plot_fold_transit(time, flux, period, epoch, depth, dur, smooth=10,
                      lw=1.5, label="Box1DSmooth")
 
         plt.legend(loc="upper right")
-        plt.xlabel('Phased Times [%s]' % units)
+        plt.xlabel('Phased Time [%s]' % units)
 
         if transit_only:
             xmin = np.min(phases[intransit])

{exovetter-0.0.9 → exovetter-0.0.11}/exovetter.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: exovetter
-Version: 0.0.9
+Version: 0.0.11
 Summary: Exoplanet vetting package
 Home-page: https://github.com/spacetelescope/exovetter
 Author: Susan Mullally et al.