From ca68eb4a4b874da50897a94ee4df9b12b7dae3b8 Mon Sep 17 00:00:00 2001 From: Casper da Costa-Luis Date: Sun, 28 Jul 2019 16:04:23 +0100 Subject: [PATCH] add progress.Tqdm.update_desc(truncate=True) --- dvc/progress.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/dvc/progress.py b/dvc/progress.py index 042c8c1be..0f5c56794 100644 --- a/dvc/progress.py +++ b/dvc/progress.py @@ -38,11 +38,13 @@ class Tqdm(tqdm): **kwargs) # self.set_lock(Lock()) - def update_desc(self, desc, n=1): + def update_desc(self, desc, n=1, truncate=True): """ - Calls `set_description(desc)` and `update(n)` + Calls `set_description(truncate(desc))` and `update(n)` """ - self.set_description(desc, refresh=False) + self.set_description( + self.truncate(desc) if truncate else desc, refresh=False + ) self.update(n) def update_to(self, current, total=None):