From 1987856470d42dbd9e93500ce4fba6d725a3f298 Mon Sep 17 00:00:00 2001
From: "Dr. Kashif Rasul" <kashif.rasul@gmail.com>
Date: Wed, 15 Jan 2020 16:59:31 +0100
Subject: [PATCH] tqdm over eval dims

---
 pts/evaluation/evaluator.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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),