Skip to content

Commit

Permalink
adding more update points and a label
Browse files Browse the repository at this point in the history
  • Loading branch information
bmeyers committed Mar 15, 2024
1 parent 130730d commit 1c7c53e
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion solardatatools/data_handler.py
Original file line number Diff line number Diff line change
Expand Up @@ -245,7 +245,7 @@ def run_pipeline(
######################################################################
t[0] = time()
if verbose:
progress = tqdm(total=5)
progress = tqdm(desc="task list", total=7)
if self.data_frame_raw is not None:
# If power_col not passed, assume that the first column contains the
# data to be processed
Expand All @@ -265,6 +265,8 @@ def run_pipeline(
)
if correct_tz:
self.tz_correction = sn_deviation
if verbose:
progress.update()
# Embed the data as a matrix, with days in columns. Also, set some
# attributes, like the scan rate, day index, and day of year arary.
# Almost never use start_day_ix and end_day_ix, but they're there
Expand Down Expand Up @@ -292,6 +294,8 @@ def run_pipeline(
self.boolean_masks.missing_values = np.isnan(self.raw_data_matrix)
# Run once to get a rough estimate. Update at the end after cleaning
# is finished
if verbose:
progress.update()
ss = SunriseSunset()
try:
ss.run_optimizer(self.raw_data_matrix, plot=False, solver=solver_convex)
Expand Down

0 comments on commit 1c7c53e

Please sign in to comment.