HLSP fix for Kepler/K2

lightkurve · Jul 12, 2024 · b367a29 · b367a29
1 parent 77d4c66
commit b367a29
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/src/lksearch/K2Search.py b/src/lksearch/K2Search.py
@@ -78,10 +78,16 @@ def __init__(
         exptime: Optional[Union[str, int, tuple]] = (0, 9999),
         pipeline: Optional[Union[str, list[str]]] = None,
         campaign: Optional[Union[int, list[int]]] = None,
+        hlsp: bool = True,
     ):
+        if hlsp is False:
+            pipeline = ["K2"]
+            self.mission_search = ["K2"]
+        else:
+            self.mission_search = ["K2", "HLSP"]
         super().__init__(
             target=target,
-            mission=["K2"],
+            mission=self.mission_search,
             obs_table=obs_table,
             prod_table=prod_table,
             table=table,

diff --git a/src/lksearch/KeplerSearch.py b/src/lksearch/KeplerSearch.py
@@ -81,10 +81,17 @@ def __init__(
         pipeline: Optional[Union[str, list[str]]] = None,
         quarter: Optional[Union[int, list[int]]] = None,
         month: Optional[int] = None,
+        hlsp: bool = True,
     ):
+        if hlsp is False:
+            pipeline = ["Kepler"]
+            self.mission_search = ["Kepler"]
+        else:
+            self.mission_search = ["Kepler", "HLSP"]
+
         super().__init__(
             target=target,
-            mission=["Kepler"],
+            mission=self.mission_search,
             obs_table=obs_table,
             prod_table=prod_table,
             table=table,