arvi 0.1.14__tar.gz → 0.1.16__tar.gz

{arvi-0.1.14 → arvi-0.1.16}/.github/workflows/python-publish.yml

@@ -22,9 +22,9 @@ jobs:
     runs-on: ubuntu-latest
 
     steps:
-    - uses: actions/checkout@v3
+    - uses: actions/checkout@v4
     - name: Set up Python
-      uses: actions/setup-python@v3
+      uses: actions/setup-python@v5
       with:
         python-version: '3.x'
     - name: Install dependencies

{arvi-0.1.14 → arvi-0.1.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: arvi
-Version: 0.1.14
+Version: 0.1.16
 Summary: The Automated RV Inspector
 Author-email: João Faria <joao.faria@unige.ch>
 License: MIT
@@ -14,11 +14,9 @@ Requires-Dist: mock; python_version < "3.3"
 Requires-Dist: numpy
 Requires-Dist: scipy
 Requires-Dist: matplotlib
-Requires-Dist: mplcursors
 Requires-Dist: astropy
 Requires-Dist: dace-query
 Requires-Dist: loguru
-Requires-Dist: mpldatacursor
 Requires-Dist: tqdm
 Requires-Dist: pySWEETCat
 Requires-Dist: kepmodel

{arvi-0.1.14 → arvi-0.1.16}/arvi/binning.py

@@ -1,8 +1,4 @@
 import numpy as np
-from numpy.testing import suppress_warnings
-
-from scipy.stats import binned_statistic as old_binned_statistic,\
-                        binned_statistic_dd as old_binned_statistic_dd
 
 from .setup_logger import logger
 
@@ -38,6 +34,7 @@ def binned_statistic(x, values, statistic='mean', bins=10, range=None,
 
 def binned_statistic_dd(sample, values, statistic='mean', bins=10, range=None,
                         expand_binnumbers=False, weights=None):
+    from numpy.testing import suppress_warnings
     known_stats = [
         'mean', 'median', 'count', 'sum', 'std', 'min', 'max', 'ptp'
     ]
@@ -214,11 +211,13 @@ def binned_statistic_dd(sample, values, statistic='mean', bins=10, range=None,
     return result, edges, binnumbers
 
 
-# put back the documentation
-doc1 = old_binned_statistic.__doc__
-doc2 = old_binned_statistic_dd.__doc__
-binned_statistic.__doc__ = doc1
-binned_statistic_dd.__doc__ = doc2
+# # put back the documentation
+# from scipy.stats import binned_statistic as old_binned_statistic,\
+#                         binned_statistic_dd as old_binned_statistic_dd
+# doc1 = old_binned_statistic.__doc__
+# doc2 = old_binned_statistic_dd.__doc__
+# binned_statistic.__doc__ = doc1
+# binned_statistic_dd.__doc__ = doc2
 
 ###############################################################################
 
@@ -339,6 +338,9 @@ def binRV(time, rv, err=None, stat='wmean', tstat='wmean', estat='addquad',
     if (err is not None) and (stat == 'wmean'):
         stat = wmean  # default is weighted mean
 
+    if (err is None) and (stat == 'wmean'):
+        stat = 'mean'
+
     brv = binned_statistic(time, rv, statistic=stat, bins=bins, range=None,
                            weights=err)
 
@@ -346,6 +348,9 @@ def binRV(time, rv, err=None, stat='wmean', tstat='wmean', estat='addquad',
     if (err is not None) and (tstat == 'wmean'):
         tstat = wmean  # default is weighted mean
 
+    if (err is None) and (tstat == 'wmean'):
+        tstat = 'mean'
+
     times = binned_statistic(time, time, statistic=tstat, bins=bins,
                              range=None, weights=err)
     # if there are errors, bin them too

{arvi-0.1.14 → arvi-0.1.16}/arvi/dace_wrapper.py

@@ -23,7 +23,7 @@ def load_spectroscopy() -> SpectroscopyClass:
 
 @lru_cache()
 def get_dace_id(star):
-    filters = {"obj_id_catname": {"contains": [star]}}
+    filters = {"obj_id_catname": {"equal": [star]}}
     try:
         with stdout_disabled(), all_logging_disabled():
             r = load_spectroscopy().query_database(filters=filters, limit=1)
@@ -36,10 +36,6 @@ def get_arrays(result, latest_pipeline=True, ESPRESSO_mode='HR11', NIRPS_mode='H
     arrays = []
     instruments = list(result.keys())
 
-    # if verbose:
-    #     if latest_pipeline:
-    #         logger.info('selecting latest pipeline version')
-
     for inst in instruments:
         pipelines = list(result[inst].keys())
 
@@ -226,11 +222,12 @@ def get_observations(star, instrument=None, main_id=None, verbose=True):
         raise ValueError(msg)
 
     # sort pipelines, being extra careful with HARPS pipeline names
-    # (i.e. ensure that 3.0.0 > 3.5)
+    # (i.e. ensure that 3.x.x > 3.5)
+    from re import match
     def cmp(a, b):
-        if a[0] in ('3.5', '3.5 EGGS') and b[0] == '3.0.0':
+        if a[0] in ('3.5', '3.5 EGGS') and match(r'3.\d.\d', b[0]):
             return -1
-        if b[0] in ('3.5', '3.5 EGGS') and a[0] == '3.0.0':
+        if b[0] in ('3.5', '3.5 EGGS') and match(r'3.\d.\d', a[0]):
             return 1
 
         if a[0] == b[0]:
@@ -243,8 +240,7 @@ def get_observations(star, instrument=None, main_id=None, verbose=True):
     from functools import cmp_to_key
     new_result = {}
     for inst in instruments:
-        new_result[inst] = dict(sorted(result[inst].items(),
-                                       key=cmp_to_key(cmp), reverse=True))
+        new_result[inst] = dict(sorted(result[inst].items(), key=cmp_to_key(cmp), reverse=True))
 
     if verbose:
         logger.info('RVs available from')

{arvi-0.1.14 → arvi-0.1.16}/arvi/gaia_wrapper.py

@@ -26,7 +26,7 @@ def run_query(query):
     url = 'https://gea.esac.esa.int/tap-server/tap/sync'
     data = dict(query=query, request='doQuery', lang='ADQL', format='csv')
     try:
-        response = requests.post(url, data=data, timeout=10)
+        response = requests.post(url, data=data, timeout=5)
     except requests.ReadTimeout as err:
         raise IndexError(err)
     except requests.ConnectionError as err:

{arvi-0.1.14 → arvi-0.1.16}/arvi/plots.py

@@ -1,12 +1,7 @@
-import os
-from functools import partial, partialmethod
+from functools import partialmethod, wraps
 from itertools import cycle
 
-import matplotlib.collections
 import numpy as np
-import matplotlib
-import matplotlib.pyplot as plt
-import mplcursors
 
 from astropy.timeseries import LombScargle
 
@@ -14,11 +9,24 @@ from .setup_logger import logger
 from . import config
 from .stats import wmean
 
+from .utils import lazy_import
+plt = lazy_import('matplotlib.pyplot')
+
+
+def plot_fast(func):
+    @wraps(func)
+    def wrapper(*args, **kwargs):
+        with plt.style.context('fast'):
+            return func(*args, **kwargs)
+    return wrapper
+
+
 
 class BlittedCursor:
     """ A cross-hair cursor using blitting for faster redraw. """
     def __init__(self, axes, vertical=True, horizontal=True, show_text=None,
                  transforms_x=None, transforms_y=None):
+        import matplotlib # delay import
         if isinstance(axes, matplotlib.axes.Axes):
             axes = [axes]
         self.axes = axes
@@ -102,10 +110,31 @@ class BlittedCursor:
                     ax.draw_artist(self.text)
                 ax.figure.canvas.blit(ax.bbox)
 
+def clickable_legend(fig, ax, leg):
+    from matplotlib.text import Text
+    handles, labels = ax.get_legend_handles_labels()
+    for text in leg.get_texts():
+        text.set_picker(True)
 
+    def on_pick_legend(event):
+        artist = event.artist
+        if isinstance(artist, Text):
+            try:
+                h = handles[labels.index(artist.get_text())]
+                alpha_text = {None:0.2, 1.0: 0.2, 0.2:1.0}[artist.get_alpha()]
+                alpha_point = {None: 0.0, 1.0: 0.0, 0.2: 1.0}[artist.get_alpha()]
+                h[0].set_alpha(alpha_point)
+                h[2][0].set_alpha(alpha_point)
+                artist.set_alpha(alpha_text)
+                fig.canvas.draw()
+            except ValueError:
+                pass
+    return on_pick_legend
+
+# @plot_fast
 def plot(self, ax=None, show_masked=False, instrument=None, time_offset=0,
-         remove_50000=False, tooltips=False, label=None, N_in_label=False,
-         versus_n=False, show_histogram=False, bw=False, **kwargs):
+         remove_50000=False, tooltips=False, show_legend=True, label=None,
+         N_in_label=False, versus_n=False, show_histogram=False, bw=False, **kwargs):
     """ Plot the RVs
 
     Args:
@@ -121,6 +150,8 @@ def plot(self, ax=None, show_masked=False, instrument=None, time_offset=0,
             Whether to subtract 50000 from time. Defaults to False.
         tooltips (bool, optional):
             Show information upon clicking a point. Defaults to True.
+        show_legend (bool, optional):
+            Show legend. Defaults to True.
         N_in_label (bool, optional):
             Show number of observations in legend. Defaults to False.
         versus_n (bool, optional):
@@ -145,12 +176,14 @@ def plot(self, ax=None, show_masked=False, instrument=None, time_offset=0,
             fig, (ax, axh) = plt.subplots(1, 2, constrained_layout=True, gridspec_kw={'width_ratios': [3, 1]})
         else:
             fig, ax = plt.subplots(1, 1, constrained_layout=True)
+    elif ax == -1:
+        ax = plt.gca()
+        fig = ax.figure
     else:
         if show_histogram:
             ax, axh = ax
         fig = ax.figure
 
-
     kwargs.setdefault('ls', '')
     kwargs.setdefault('capsize', 0)
     kwargs.setdefault('ms', 4)
@@ -172,7 +205,6 @@ def plot(self, ax=None, show_masked=False, instrument=None, time_offset=0,
     except AttributeError:
         zorders = cycle([1] * len(instruments))
 
-    cursors = {}
     containers = {}
     for _i, inst in enumerate(instruments):
         s = self if self._child else getattr(self, inst)
@@ -207,22 +239,6 @@ def plot(self, ax=None, show_masked=False, instrument=None, time_offset=0,
             hlabel = f'{s.mvrad.std():.2f} {self.units}'
             axh.hist(s.mvrad, **kw, label=hlabel)
 
-                # cursors[inst] = crsr = mplcursors.cursor(container, multiple=False)
-                # @crsr.connect("add")
-                # def _(sel):
-                #     inst = sel.artist.get_label()
-                #     _s = getattr(self, inst)
-                #     vrad, svrad = _s.vrad[sel.index], _s.svrad[sel.index]
-                #     sel.annotation.get_bbox_patch().set(fc="white")
-                #     text = f'{inst}\n'
-                #     text += f'BJD: {sel.target[0]:9.5f}\n'
-                #     text += f'RV: {vrad:.3f} ± {svrad:.3f}'
-                #     # if fig.canvas.manager.toolmanager.get_tool('infotool').toggled:
-                #     #     text += '\n\n'
-                #     #     text += f'date: {_s.date_night[sel.index]}\n'
-                #     #     text += f'mask: {_s.ccf_mask[sel.index]}'
-                #     sel.annotation.set_text(text)
-
         if show_masked:
             if versus_n:
                 pass
@@ -239,62 +255,104 @@ def plot(self, ax=None, show_masked=False, instrument=None, time_offset=0,
             ax.errorbar(self.time[~self.mask] - time_offset, self.vrad[~self.mask], self.svrad[~self.mask],
                         label='masked', fmt='x', ms=10, color='k', zorder=-2)
 
-    leg = ax.legend()
-    handles, labels = ax.get_legend_handles_labels()
-    for text in leg.get_texts():
-        text.set_picker(True)
-
-    def on_pick_legend(event):
-        artist = event.artist
-        if isinstance(artist, matplotlib.text.Text):
-            try:
-                h = handles[labels.index(artist.get_text())]
-                alpha_text = {None:0.2, 1.0: 0.2, 0.2:1.0}[artist.get_alpha()]
-                alpha_point = {None: 0.0, 1.0: 0.0, 0.2: 1.0}[artist.get_alpha()]
-                h[0].set_alpha(alpha_point)
-                h[2][0].set_alpha(alpha_point)
-                artist.set_alpha(alpha_text)
-                fig.canvas.draw()
-            except ValueError:
-                pass
-    plt.connect('pick_event', on_pick_legend)
+    if show_legend:
+        leg = ax.legend()
+        on_pick_legend = clickable_legend(fig, ax, leg)
+        plt.connect('pick_event', on_pick_legend)
 
     if tooltips:
+        from matplotlib.lines import Line2D
+        from matplotlib.collections import LineCollection
         annotations = []
-        def on_pick_point(event):
-            print('annotations:', annotations)
+        axes_artists = []
+        selected_inds = []
+        selected_insts = []
+
+        def _cleanup():
             for text in annotations:
                 text.remove()
                 annotations.remove(text)
+            for art in axes_artists:
+                art.remove()
+                axes_artists.remove(art)
+            for ind in selected_inds:
+                selected_inds.remove(ind)
+
+        def on_press(event):
+            # print('press', event.key)
+            if event.key in ('r',):
+                for i, inst in zip(selected_inds, selected_insts):
+                    i = self._index_from_instrument_index(i, inst)
+                    self.remove_point(i)
+                _cleanup()
+                fig.canvas.draw_idle()
+            if event.key == 'escape':
+                _cleanup()
+                kp_cid = None
+                for k, _f in fig.canvas.callbacks.callbacks['key_press_event'].items():
+                    if _f._obj == on_press:
+                        kp_cid = k
+                if kp_cid is not None:
+                    fig.canvas.callbacks.disconnect(k)
+                fig.canvas.draw_idle()
 
+        def on_pick_point(event):
+            _cleanup()            
             artist = event.artist
-            if isinstance(artist, (matplotlib.lines.Line2D, matplotlib.collections.LineCollection)):
-                print(event.ind, artist)
-                if isinstance(artist, matplotlib.lines.Line2D):
+            if isinstance(artist, (Line2D, LineCollection)):
+                # print(event.ind, artist)
+                if isinstance(artist, Line2D):
+                    ind = event.ind
                     matching_instrument = [k for k, v in containers.items() if artist in v]
-                    print(matching_instrument)
                     if len(matching_instrument) == 0:
                         return
                     inst = matching_instrument[0]
+
                     _s = getattr(self, inst)
-                    ind = event.ind[0]
-                    # print(_s.mtime[ind], _s.mvrad[ind], _s.msvrad[ind])
-
-                    text = f'{inst}\n'
-                    text += f'{_s.mtime[ind]:9.5f}\n'
-                    text += f'RV: {_s.mvrad[ind]:.1f} ± {_s.msvrad[ind]:.1f}'
-
-                    annotations.append(
-                        ax.annotate(text, (_s.mtime[ind], _s.mvrad[ind]), xycoords='data',
-                                    xytext=(5, 10), textcoords='offset points', fontsize=9,
-                                    bbox={'boxstyle': 'round', 'fc': 'w'}, arrowprops=dict(arrowstyle="-"))
-                    )
+
+                    if event.ind.size > 1:
+                        mint, maxt = _s.mtime[ind].min(), _s.mtime[ind].max()
+                        miny, maxy = _s.mvrad[ind].min(), _s.mvrad[ind].max()
+                        axins = ax.inset_axes([0.1, 0.5, 0.5, 0.4],
+                            xlim=(mint - 0.1 * (maxt - mint), maxt + 0.1 * (maxt - mint)), 
+                            ylim=(miny - 0.1 * (maxy - miny), maxy + 0.1 * (maxy - miny)), 
+                            xticklabels=[], yticklabels=[]
+                            )
+                        axins.errorbar(_s.mtime[ind], _s.mvrad[ind], _s.msvrad[ind], fmt='o', ms=3,
+                                       color=artist.get_color())
+                        axins.margins(x=0.5)
+                        axins.autoscale_view()
+                        rectangle_patch, connector_lines = ax.indicate_inset_zoom(axins, edgecolor="black")
+                        axes_artists.append(axins)
+                        axes_artists.append(rectangle_patch)
+                        for line in connector_lines:
+                            axes_artists.append(line)
+                    else:
+                        ind = event.ind[0]
+                        selected_inds.append(ind)
+                        selected_insts.append(inst)
+                        # print(_s.mtime[ind], _s.mvrad[ind], _s.msvrad[ind])
+
+                        text = f'{inst} ({ind})\n'
+                        text += f'{_s.mtime[ind]:9.5f}\n'
+                        text += f'RV: {_s.mvrad[ind]:.1f} ± {_s.msvrad[ind]:.1f}'
+
+                        annotations.append(
+                            ax.annotate(text, (_s.mtime[ind], _s.mvrad[ind]), xycoords='data',
+                                        xytext=(5, 10), textcoords='offset points', fontsize=9,
+                                        bbox={'boxstyle': 'round', 'fc': 'w'}, arrowprops=dict(arrowstyle="-"))
+                        )
                     # ax.annotate(f'{inst}', (0.5, 0.5), xycoords=artist, ha='center', va='center')
                     fig.canvas.draw()
                     # print(event.ind, artist.get_label())
+
+                    _ = fig.canvas.mpl_connect('key_press_event', on_press)
+
         plt.connect('pick_event', on_pick_point)
 
 
+
+
     if show_histogram:
         axh.legend()
 
@@ -309,25 +367,23 @@ def plot(self, ax=None, show_masked=False, instrument=None, time_offset=0,
         else:
             ax.set_xlabel('BJD - 2400000 [days]')
 
-    from matplotlib.backend_tools import ToolBase, ToolToggleBase
-    tm = fig.canvas.manager.toolmanager
-
-    class InfoTool(ToolToggleBase):
-        description = "Show extra information about each observation"
-        image = os.path.abspath(os.path.join(os.path.dirname(__file__), 'data', 'info.svg'))
-        # def enable(self, *args, **kwargs):
-        #     self.figure.add_axes([1, 0, 0.3, 1])
-        #     self.figure.canvas.draw_idle()
-
-    from PIL import UnidentifiedImageError
-    try:
-        tm.add_tool("infotool", InfoTool)
-        fig.canvas.manager.toolbar.add_tool(tm.get_tool("infotool"), "toolgroup")
-        raise UnidentifiedImageError
-    except AttributeError:
-        pass
-    except UnidentifiedImageError:
-        pass
+    # from matplotlib.backend_tools import ToolBase, ToolToggleBase
+    # tm = fig.canvas.manager.toolmanager
+    # class InfoTool(ToolToggleBase):
+    #     description = "Show extra information about each observation"
+    #     image = os.path.abspath(os.path.join(os.path.dirname(__file__), 'data', 'info.svg'))
+    #     # def enable(self, *args, **kwargs):
+    #     #     self.figure.add_axes([1, 0, 0.3, 1])
+    #     #     self.figure.canvas.draw_idle()
+    # from PIL import UnidentifiedImageError
+    # try:
+    #     tm.add_tool("infotool", InfoTool)
+    #     fig.canvas.manager.toolbar.add_tool(tm.get_tool("infotool"), "toolgroup")
+    #     raise UnidentifiedImageError
+    # except AttributeError:
+    #     pass
+    # except UnidentifiedImageError:
+    #     pass
 
 
     if config.return_self:
@@ -339,8 +395,9 @@ def plot(self, ax=None, show_masked=False, instrument=None, time_offset=0,
     return fig, ax
 
 
+@plot_fast
 def plot_quantity(self, quantity, ax=None, show_masked=False, instrument=None,
-                  time_offset=0, remove_50000=False, tooltips=False,
+                  time_offset=0, remove_50000=False, tooltips=False, show_legend=True,
                   N_in_label=False, **kwargs):
     if self.N == 0:
         if self.verbose:
@@ -399,17 +456,21 @@ def plot_quantity(self, quantity, ax=None, show_masked=False, instrument=None,
                     getattr(self, quantity + '_err')[~self.mask],
                     label='masked', fmt='x', ms=10, color='k', zorder=-2)
 
-    ax.legend()
+    if show_legend:
+        leg = ax.legend()
+        on_pick_legend = clickable_legend(fig, ax, leg)
+        plt.connect('pick_event', on_pick_legend)
+
     ax.minorticks_on()
 
     ylabel = {
+        quantity: quantity,
         'fwhm': f'FWHM [{self.units}]',
         'bispan': f'BIS [{self.units}]',
         'rhk': r"$\log$ R'$_{HK}$",
-        'berv': f'BERV [km/s]',
-        quantity: quantity,
+        'berv': 'BERV [km/s]',
     }
-    ax.set_ylabel(ylabel[quantity])
+    ax.set_ylabel(ylabel[quantity.lower()])
 
     if remove_50000:
         ax.set_xlabel('BJD - 2450000 [days]')
@@ -428,6 +489,7 @@ plot_rhk = partialmethod(plot_quantity, quantity='rhk')
 plot_berv = partialmethod(plot_quantity, quantity='berv')
 
 
+@plot_fast
 def gls(self, ax=None, label=None, fap=True, instrument=None, adjust_means=config.adjust_means_gls,
         picker=True, **kwargs):
     """
@@ -471,7 +533,7 @@ def gls(self, ax=None, label=None, fap=True, instrument=None, adjust_means=confi
         if self.verbose:
             logger.info(f'calculating periodogram for instrument {instrument}')
 
-        if adjust_means:
+        if adjust_means and not self._child:
             if self.verbose:
                 logger.info('adjusting instrument means before gls')
             means = np.empty_like(y)
@@ -486,7 +548,7 @@ def gls(self, ax=None, label=None, fap=True, instrument=None, adjust_means=confi
         y = self.vrad[self.mask].copy()
         e = self.svrad[self.mask].copy()
 
-        if adjust_means:
+        if adjust_means and not self._child:
             if self.verbose:
                 logger.info('adjusting instrument means before gls')
             means = np.empty_like(y)
@@ -529,6 +591,7 @@ def gls(self, ax=None, label=None, fap=True, instrument=None, adjust_means=confi
         ax.legend()
 
     if ax.get_legend() is not None:
+        from matplotlib.text import Text
         leg = ax.get_legend()
         for text in leg.get_texts():
             text.set_picker(True)
@@ -536,7 +599,7 @@ def gls(self, ax=None, label=None, fap=True, instrument=None, adjust_means=confi
         def on_pick_legend(event):
             handles, labels = ax.get_legend_handles_labels()
             artist = event.artist
-            if isinstance(artist, matplotlib.text.Text):
+            if isinstance(artist, Text):
                 # print('handles:', handles)
                 # print('labels:', labels)
                 # print(artist.get_text())
@@ -560,6 +623,7 @@ def gls(self, ax=None, label=None, fap=True, instrument=None, adjust_means=confi
         return fig, ax
 
 
+@plot_fast
 def gls_quantity(self, quantity, ax=None, fap=True, picker=True):
     if not hasattr(self, quantity):
         logger.error(f"cannot find '{quantity}' attribute")
@@ -572,7 +636,10 @@ def gls_quantity(self, quantity, ax=None, fap=True, picker=True):
 
     t = self.mtime
     y = getattr(self, quantity)[self.mask]
-    ye = getattr(self, quantity + '_err')[self.mask]
+    try:
+        ye = getattr(self, quantity + '_err')[self.mask]
+    except AttributeError:
+        ye = None
 
     if np.isnan(y).any():
         if self.verbose:

{arvi-0.1.14 → arvi-0.1.16}/arvi/reports.py

@@ -1,9 +1,6 @@
 from functools import partial
 import numpy as np
 from astropy.timeseries import LombScargle
-import matplotlib.pyplot as plt
-import matplotlib.gridspec as gridspec
-from matplotlib.backends.backend_pdf import PdfPages
 
 from .setup_logger import logger
 
@@ -31,6 +28,10 @@ def sine_picker(event, self, fig, ax, ax1):
 
 
 def report(self, save=None):
+    import matplotlib.pyplot as plt
+    import matplotlib.gridspec as gridspec
+    from matplotlib.backends.backend_pdf import PdfPages
+
     # size = A4
     size = 8.27, 11.69
     fig = plt.figure(figsize=size, constrained_layout=True)

{arvi-0.1.14 → arvi-0.1.16}/arvi/simbad_wrapper.py

@@ -1,7 +1,6 @@
 import os
 import requests
 
-from astropy.coordinates import SkyCoord
 import pysweetcat
 
 DATA_PATH = os.path.dirname(__file__)
@@ -44,7 +43,7 @@ def run_query(query):
     url = 'http://simbad.u-strasbg.fr/simbad/sim-tap/sync'
     data = dict(query=query, request='doQuery', lang='ADQL', format='text/plain', phase='run')
     try:
-        response = requests.post(url, data=data, timeout=10)
+        response = requests.post(url, data=data, timeout=5)
     except requests.ReadTimeout as err:
         raise IndexError(err)
     except requests.ConnectionError as err:
@@ -97,6 +96,8 @@ class simbad:
         Args:
             star (str): The name of the star to query simbad
         """
+        from astropy.coordinates import SkyCoord
+
         self.star = star
 
         if 'kobe' in self.star.lower():

{arvi-0.1.14 → arvi-0.1.16}/arvi/stats.py

@@ -1,7 +1,4 @@
 import numpy as np
-from scipy.stats import median_abs_deviation
-from scipy.stats._stats_py import SigmaclipResult
-
 
 def wmean(a, e):
     """Weighted mean of array `a`, with uncertainties given by `e`.
@@ -69,6 +66,8 @@ def sigmaclip_median(a, low=4.0, high=4.0):
             - `lower`: Lower clipping limit
             - `upper`: Upper clipping limit
     """
+    from scipy.stats import median_abs_deviation
+    from scipy.stats._stats_py import SigmaclipResult
     c = np.asarray(a).ravel()
     delta = 1
     while delta:

{arvi-0.1.14 → arvi-0.1.16}/arvi/utils.py

@@ -122,6 +122,19 @@ def find_data_file(file):
 
     return data_file
 
+
+import importlib.util
+import sys
+def lazy_import(name):
+    spec = importlib.util.find_spec(name)
+    loader = importlib.util.LazyLoader(spec.loader)
+    spec.loader = loader
+    module = importlib.util.module_from_spec(spec)
+    sys.modules[name] = module
+    loader.exec_module(module)
+    return module
+
+
 def ESPRESSO_ADC_issues():
     adc_file = find_data_file('obs_affected_ADC_issues.dat')
     lines = [line.strip() for line in open(adc_file).readlines()]

{arvi-0.1.14 → arvi-0.1.16}/arvi.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: arvi
-Version: 0.1.14
+Version: 0.1.16
 Summary: The Automated RV Inspector
 Author-email: João Faria <joao.faria@unige.ch>
 License: MIT
@@ -14,11 +14,9 @@ Requires-Dist: mock; python_version < "3.3"
 Requires-Dist: numpy
 Requires-Dist: scipy
 Requires-Dist: matplotlib
-Requires-Dist: mplcursors
 Requires-Dist: astropy
 Requires-Dist: dace-query
 Requires-Dist: loguru
-Requires-Dist: mpldatacursor
 Requires-Dist: tqdm
 Requires-Dist: pySWEETCat
 Requires-Dist: kepmodel

{arvi-0.1.14 → arvi-0.1.16}/arvi.egg-info/requires.txt

@@ -1,11 +1,9 @@
 numpy
 scipy
 matplotlib
-mplcursors
 astropy
 dace-query
 loguru
-mpldatacursor
 tqdm
 pySWEETCat
 kepmodel

{arvi-0.1.14 → arvi-0.1.16}/pyproject.toml

@@ -8,7 +8,7 @@ authors = [
     {name = "João Faria", email = "joao.faria@unige.ch"},
 ]
 description = "The Automated RV Inspector"
-version = "0.1.14"
+version = "0.1.16"
 readme = {file = "README.md", content-type = "text/markdown"}
 requires-python = ">=3.8"
 keywords = ["RV", "exoplanets"]
@@ -22,11 +22,9 @@ dependencies = [
     "numpy",
     "scipy",
     "matplotlib",
-    "mplcursors",
     "astropy",
     "dace-query",
     "loguru",
-    "mpldatacursor",
     "tqdm",
     "pySWEETCat",
     "kepmodel",