Skip to content

Commit

Permalink
Adding stub classes to LightCurve that child classes must implement.
Browse files Browse the repository at this point in the history
  • Loading branch information
drewoldag committed Oct 31, 2024
1 parent c683c9b commit 3213bb6
Showing 1 changed file with 22 additions and 0 deletions.
22 changes: 22 additions & 0 deletions src/resspect/feature_extractors/light_curve.py
Original file line number Diff line number Diff line change
Expand Up @@ -121,6 +121,28 @@ def __init__(self):
self.sncode = 0
self.sntype = ' '

def fit(self, band: str) -> np.ndarray:
"""
Extract features for one filter.
Parameters
----------
band: str
Choice of broad band filter
Returns
-------
np.ndarray
"""
raise NotImplementedError()

def fit_all(self):
"""
Performs feature extraction for all filters independently and concatenate results.
Populates self.features.
"""
raise NotImplementedError()

def _get_snpcc_photometry_raw_and_header(
self, lc_data: np.ndarray,
sntype_test_value: str = "-9") -> Tuple[np.ndarray, list]:
Expand Down

0 comments on commit 3213bb6

Please sign in to comment.