Skip to content

Commit

Permalink
fixed origin_file argument issue
Browse files Browse the repository at this point in the history
  • Loading branch information
megbedell committed Jan 23, 2019
1 parent 1082c38 commit 041cd58
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions wobble/data.py
Original file line number Diff line number Diff line change
Expand Up @@ -53,7 +53,7 @@ def __init__(self, filename, filepath='',
if np.sum(orders_to_cut) > 0:
print("Data: Dropping orders {0} because they have average SNR < {1:.0f}".format(orders[orders_to_cut], min_snr))
orders = orders[~orders_to_cut]
self.read_data(orders=orders, epochs=epochs) # overwrite with new data
self.read_data(origin_file, orders=orders, epochs=epochs) # overwrite with new data
self.mask_low_pixels(min_flux=min_flux, padding=padding, min_snr=min_snr)
if len(orders) == 0:
print("All orders failed the quality cuts with min_snr={0:.0f}.".format(min_snr))
Expand All @@ -64,7 +64,7 @@ def __init__(self, filename, filepath='',
if np.sum(epochs_to_cut) > 0:
print("Data: Dropping epochs {0} because they have average SNR < {1:.0f}".format(epochs[epochs_to_cut], min_snr))
epochs = epochs[~epochs_to_cut]
self.read_data(orders=orders, epochs=epochs) # overwrite with new data
self.read_data(origin_file, orders=orders, epochs=epochs) # overwrite with new data
self.mask_low_pixels(min_flux=min_flux, padding=padding, min_snr=min_snr)
if len(epochs) == 0:
print("All epochs failed the quality cuts with min_snr={0:.0f}.".format(min_snr))
Expand Down

0 comments on commit 041cd58

Please sign in to comment.