update __repr__; fixes quality key
@@ -741,7 +741,10 @@ class KeplerLightCurveFile(LightCurveFile):
return KeplerLightCurveFile(path)
def __repr__(self):
- return('KeplerLightCurveFile Object (ID: {})'.format(self.keplerid))
+ if self.mission.lower() == 'kepler':
+ return('KeplerLightCurveFile(KIC: {})'.format(self.keplerid))
+ elif self.mission.lower() == 'k2':
+ return('KeplerLightCurveFile(EPIC: {})'.format(self.keplerid))
def get_lightcurve(self, flux_type, centroid_type='MOM_CENTR'):
if flux_type in self._flux_types():
@@ -859,7 +862,7 @@ class TessLightCurveFile(LightCurveFile):
self.quality_mask = self._quality_mask(quality_bitmask)
def __repr__(self):
- return('TessLightCurveFile Object (ID: {})'.format(self.keplerid))
+ return('TessLightCurveFile(TICID: {})'.format(self.keplerid))
@property
def ticid(self):
@@ -872,10 +875,10 @@ class TessLightCurveFile(LightCurveFile):
flux_err=self.hdu[1].data[flux_type + "_ERR"][self.quality_mask],
centroid_col=self.hdu[1].data[centroid_type + "1"][self.quality_mask],
centroid_row=self.hdu[1].data[centroid_type + "2"][self.quality_mask],
- quality=self.hdu[1].data['SAP_QUALITY'][self.quality_mask],
+ quality=self.hdu[1].data['QUALITY'][self.quality_mask],
quality_bitmask=self.quality_bitmask,
cadenceno=self.cadenceno,
- ticid=self.keplerid)
+ ticid=self.ticid)
class SFFCorrector(object):
