Skip to content

Commit ecd3504

Browse files
author
Cody Rude
committed
Fixed bug in kepler data fetcher where it was only pre-processing the
last lightcurve
1 parent 3e5a07a commit ecd3504

1 file changed

Lines changed: 9 additions & 8 deletions

File tree

skdaccess/astro/kepler/data_fetcher.py

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -120,14 +120,15 @@ def output(self):
120120

121121

122122
for quarter in quarter_list:
123-
data = kid_data[kid].loc[kid_data[kid]['QUARTER'] == quarter, 'PDCSAP_FLUX']
124-
if len(data) > 0:
125-
if self.normalize:
126-
kid_data[kid].loc[kid_data[kid]['QUARTER'] == quarter, 'PDCSAP_FLUX'] = data / np.nanmedian(data)
127-
128-
if self.filter_window != None:
129-
data = kid_data[kid][kid_data[kid]['QUARTER'] == quarter].loc[:,'PDCSAP_FLUX']
130-
kid_data[kid].update(data - trend_util.medianFilter(data, self.filter_window, interpolate=False))
123+
for kid in kid_list:
124+
data = kid_data[kid].loc[kid_data[kid]['QUARTER'] == quarter, 'PDCSAP_FLUX']
125+
if len(data) > 0:
126+
if self.normalize:
127+
kid_data[kid].loc[kid_data[kid]['QUARTER'] == quarter, 'PDCSAP_FLUX'] = data / np.nanmedian(data)
128+
129+
if self.filter_window != None:
130+
data = kid_data[kid][kid_data[kid]['QUARTER'] == quarter].loc[:,'PDCSAP_FLUX']
131+
kid_data[kid].update(data - trend_util.medianFilter(data, self.filter_window, interpolate=False))
131132

132133
if self.wrapper_type == 'series':
133134
return DataWrapper(kid_data)

0 commit comments

Comments
 (0)