Skip to content

Commit

Permalink
tweak progress bar
Browse files Browse the repository at this point in the history
  • Loading branch information
gavinmacaulay committed Dec 16, 2024
1 parent f713f96 commit f1f49ea
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions src/echosms/scattermodelbase.py
Original file line number Diff line number Diff line change
Expand Up @@ -136,9 +136,8 @@ def calculate_ts(self, data, expand=False, inplace=False, multiprocess=False, pr
ts = mapply(data_df, self.__ts_helper, args=(p,), axis=1)
else: # this uses just one CPU
if progress:
tqdm.pandas(desc=self.short_name,
bar_format='{l_bar}{bar} [{n_fmt}/{total_fmt}; {rate_noinv_fmt}]',
unit='model')
tqdm.pandas(desc=self.short_name, unit=' models',
bar_format='{l_bar}{bar} [{n_fmt}/{total_fmt}; {rate_noinv_fmt}]')
ts = data_df.progress_apply(self.__ts_helper, args=(p,), axis=1)
else:
ts = data_df.apply(self.__ts_helper, args=(p,), axis=1)
Expand Down

0 comments on commit f1f49ea

Please sign in to comment.