improve progressbar

w4k2 · Aug 16, 2022 · ad6d4f2 · ad6d4f2
1 parent 368b985
commit ad6d4f2
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/weles/evaluation/Evaluator.py b/weles/evaluation/Evaluator.py
@@ -59,11 +59,10 @@ def process(self, clfs, verbose=False):
         self.true_values = np.zeros([len(self.datasets), self.m * self.k],
                                     dtype=object)
 
+        bar = tqdm(self.datasets, disable=not verbose)
         # Iterate over datasets
-        for dataset_id, dataset_name in enumerate(tqdm(self.datasets,
-                                                       desc="DTS",
-                                                       ascii=True,
-                                                       disable=not verbose)):
+        for dataset_id, dataset_name in enumerate(bar):
+            bar.set_description(dataset_name.ljust(20))
             X, y = self.datasets[dataset_name]
             for fold_id, (train, test) in enumerate(skf.split(X, y)):
                 str_gt = self._storage_key_gt(X, y, fold_id)