Skip to content

Commit

Permalink
Update docstrings
Browse files Browse the repository at this point in the history
  • Loading branch information
astrojoni89 committed Sep 9, 2023
1 parent 9dd9ee2 commit b9ea7a1
Showing 1 changed file with 16 additions and 0 deletions.
16 changes: 16 additions & 0 deletions astrosaber/parallel_processing.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,18 +22,34 @@ def init(mp_info : List):
mp_ilist = np.arange(len(mp_data))

def single_cost_i(i : int) -> Tuple[float, float, float]:
'''
Compute the cost of a single baseline spectrum.
'''
result = saberTraining.single_cost(mp_params[0], i)
return result

def lambda_extraction_i(i : int) -> Tuple[np.ndarray, np.ndarray, np.ndarray, float, float]:
'''
Generate the test data for a single spectrum.
'''
result = saberPrepare.two_step_extraction_prepare(mp_params[0], i)
return result

def two_step_i(i : int) -> Tuple[int, np.ndarray, np.ndarray, int, int]:
'''
Compute the baseline etc. for a single spectrum using the two-phase extraction.
'''
result = HisaExtraction.two_step_extraction_single(mp_params[0], i)
return result

def one_step_i(i : int) -> Tuple[int, np.ndarray, np.ndarray, int, int]:
'''
Compute the baseline etc. for a single spectrum using the one-phase extraction.
'''
result = HisaExtraction.one_step_extraction_single(mp_params[0], i)
return result

Expand Down

0 comments on commit b9ea7a1

Please sign in to comment.