diff --git a/lcviz/plugins/photometric_extraction/photometric_extraction.py b/lcviz/plugins/photometric_extraction/photometric_extraction.py index bc1916e..2f2e639 100644 --- a/lcviz/plugins/photometric_extraction/photometric_extraction.py +++ b/lcviz/plugins/photometric_extraction/photometric_extraction.py @@ -4,10 +4,6 @@ from jdaviz.core.registries import tray_registry from jdaviz.configs.cubeviz.plugins import SpectralExtraction -from jdaviz.core.template_mixin import (PluginTemplateMixin, - DatasetSelectMixin, AddResultsMixin, - skip_if_no_updates_since_last_active, - with_spinner, with_temp_disable) from jdaviz.core.user_api import PluginUserApi @@ -36,10 +32,9 @@ class PhotometricExtraction(SpectralExtraction): def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) - self.docs_link = f"https://lcviz.readthedocs.io/en/{self.vdocs}/plugins.html#photometric-extraction" + self.docs_link = f"https://lcviz.readthedocs.io/en/{self.vdocs}/plugins.html#photometric-extraction" # noqa self.docs_description = "Extract light curve from target pixel file cube." # noqa - def is_tpf(data): return len(data.shape) == 3 self.dataset.filters = [is_tpf] @@ -48,7 +43,7 @@ def is_tpf(data): @property def user_api(self): expose = ['dataset', 'function', 'aperture', - 'background', + 'background', 'add_results', 'extract', 'aperture_method'] @@ -75,7 +70,8 @@ def _return_extracted(self, cube, wcs, collapsed_nddata): return lc def _preview_x_from_extracted(self, extracted): - return extracted.time.value - self.dataset.selected_obj.meta.get('reference_time', 0.0 * u.d).value + return extracted.time.value - self.dataset.selected_obj.meta.get('reference_time', + 0.0 * u.d).value def _preview_y_from_extracted(self, extracted): return extracted.flux.value