Skip to content

Commit bd81e76

Browse files
committed
CI/BUG: Further quiet tqdm
1 parent 21a0f3f commit bd81e76

File tree

2 files changed

+7
-4
lines changed

2 files changed

+7
-4
lines changed

backtesting/backtesting.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -1308,7 +1308,8 @@ def run(self, **kwargs) -> pd.Series:
13081308
# np.nan >= 3 is not invalid; it's False.
13091309
with np.errstate(invalid='ignore'):
13101310

1311-
for i in _tqdm(range(start, len(self._data)), desc=self.run.__qualname__):
1311+
for i in _tqdm(range(start, len(self._data)), desc=self.run.__qualname__,
1312+
unit='bar', mininterval=2, miniters=100):
13121313
# Prepare data and indicators for `next` call
13131314
data._set_length(i + 1)
13141315
for attr, indicator in indicator_attrs:
@@ -1564,7 +1565,8 @@ def memoized_run(tup):
15641565
stats = self.run(**dict(tup))
15651566
return -maximize(stats)
15661567

1567-
progress = iter(_tqdm(repeat(None), total=max_tries, leave=False, desc='Backtest.optimize'))
1568+
progress = iter(_tqdm(repeat(None), total=max_tries, leave=False,
1569+
desc=self.optimize.__qualname__, mininterval=2))
15681570
_names = tuple(kwargs.keys())
15691571

15701572
def objective_function(x):

backtesting/lib.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -577,7 +577,8 @@ def run(self, **kwargs):
577577
((df_batch, self._strategy, self._bt_kwargs, kwargs)
578578
for df_batch in _batch(shm))),
579579
total=len(shm),
580-
desc=self.__class__.__name__,
580+
desc=self.run.__qualname__,
581+
mininterval=2
581582
)
582583
df = pd.DataFrame(list(chain(*results))).transpose()
583584
return df
@@ -605,7 +606,7 @@ def optimize(self, **kwargs) -> pd.DataFrame:
605606
"""
606607
heatmaps = []
607608
# Simple loop since bt.optimize already does its own multiprocessing
608-
for df in _tqdm(self._dfs, desc=self.__class__.__name__):
609+
for df in _tqdm(self._dfs, desc=self.__class__.__name__, mininterval=2):
609610
bt = Backtest(df, self._strategy, **self._bt_kwargs)
610611
_best_stats, heatmap = bt.optimize( # type: ignore
611612
return_heatmap=True, return_optimization=False, **kwargs)

0 commit comments

Comments
 (0)