tkmatrix 0.8.4__tar.gz → 0.8.6__tar.gz

{tkmatrix-0.8.4/tkmatrix.egg-info → tkmatrix-0.8.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tkmatrix
-Version: 0.8.4
+Version: 0.8.6
 Summary: ToolKit for Multi-phAse Transits Recovery from Injected eXoplanets
 Home-page: https://github.com/PlanetHunters/tkmatrix
 Author: M. Dévora-Pajares & F.J. Pozuelos

{tkmatrix-0.8.4 → tkmatrix-0.8.6}/setup.py

@@ -2,7 +2,7 @@ import setuptools
 
 with open("README.md", "r") as fh:
     long_description = fh.read()
-version = "0.8.4"
+version = "0.8.6"
 setuptools.setup(
     name="tkmatrix",
     version=version,

{tkmatrix-0.8.4 → tkmatrix-0.8.6}/tkmatrix/custom_algorithms/BlsCustomSearchAlgorithm.py

@@ -42,7 +42,7 @@ class BlsCustomSearchAlgorithm(CustomSearchAlgorithm):
                                                                                   oversampling, star_info,
                                                                                   transits_min_count)
             lc = lightkurve.LightCurve(time=time, flux=flux)
-            results = lc.to_periodogram(method='bls', period=tls_period_grid, frequency_factor=oversampling)
+            results = lc.to_periodogram(method='bls', period=tls_period_grid, frequency_factor=oversampling * 100)
             max_power_index = np.argmax(results.power)
             sde = results.power[max_power_index].value / np.nanmedian(results.power).value
             t0 = results.transit_time_at_max_power.value

{tkmatrix-0.8.4 → tkmatrix-0.8.6}/tkmatrix/properties.yaml

@@ -53,8 +53,7 @@ SIGNAL_SELECTION_MODE: 'period-epoch'
 # The SNR limit for a TLS found signal to be considered as not recovered.
 SNR_THRESHOLD: 5
 # The density of the period grid for the TLS search. Increasing this will enhance the TLS accuracy sometimes but it
-# adds a high increase of computational cost. When FIT_METHOD: bls-periodogram, OVERSAMPLING represents the
-# frequency_factor and hence, increasing it will reduce the period grid size (as opposite to the standard search).
+# adds a high increase of computational cost.
 OVERSAMPLING: 1
 # An array of time tuples to be masked in your curve. E.g: [[1401.25, 1401.60], [1415.2, 1416.0]]
 INITIAL_MASK:

{tkmatrix-0.8.4 → tkmatrix-0.8.6}/tkmatrix/tkmatrix_class.py

@@ -498,7 +498,7 @@ class MATRIX:
 
     @staticmethod
     def plot_results(object_id, inject_dir, period_grid=None, radius_grid=None, binning=1, xticks=None, yticks=None,
-                     period_grid_geom="lin", radius_grid_geom="lin", is_rv=False):
+                     period_grid_geom="lin", radius_grid_geom="lin", is_rv=False, planets_df=None):
         """
         Generates a heat map with the found/not found results for the period/radius grids
 
@@ -512,6 +512,7 @@ class MATRIX:
         :param period_grid_geom: [lin|log]
         :param radius_grid_geom: [lin|log]
         :param is_rv: whether to load the rv results or the photometry ones
+        :param planets_df: pandas dataframe with radius, period, name, radius_err_up, radius_err_bottom
         """
         filename = '/a_tls_report.csv' if not is_rv else '/a_rv_report.csv'
         column = 'radius' if not is_rv else 'mass'
@@ -570,6 +571,11 @@ class MATRIX:
         if yticks is not None:
             plt.xticks(yticks)
         ax.tick_params(axis='both', which='major', labelsize=14)
+        if planets_df is not None:
+            for index, row in planets_df.iterrows():
+                ax.errorbar([row['period']], [row['radius']],
+                            yerr=[np.full(1, row['radius_err_bottom']), np.full(1, row['radius_err_up'])],
+                            fmt='o', color='firebrick', markersize=12)
         plt.savefig(inject_dir + '/inj-rec' + ('-rv' if is_rv else '') + '.png', bbox_inches='tight', dpi=200)
         plt.close()
 

{tkmatrix-0.8.4 → tkmatrix-0.8.6/tkmatrix.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tkmatrix
-Version: 0.8.4
+Version: 0.8.6
 Summary: ToolKit for Multi-phAse Transits Recovery from Injected eXoplanets
 Home-page: https://github.com/PlanetHunters/tkmatrix
 Author: M. Dévora-Pajares & F.J. Pozuelos