diff --git a/pts/evaluation/evaluator.py b/pts/evaluation/evaluator.py index 4f91db8..5c95da6 100644 --- a/pts/evaluation/evaluator.py +++ b/pts/evaluation/evaluator.py @@ -10,6 +10,7 @@ from typing import ( Union, Callable, ) +from tqdm import tqdm # Third-party imports import numpy as np @@ -544,7 +545,7 @@ class MultivariateEvaluator(Evaluator): fcst_iterator, target_dimensionality + len(self.target_agg_funcs) ) - for dim in eval_dims: + for dim in tqdm(eval_dims): agg_metrics, metrics_per_ts = super(MultivariateEvaluator, self).__call__( self.extract_target_by_dim(ts_iterator_set[dim], dim), self.extract_forecast_by_dim(fcst_iterator_set[dim], dim),