logging: Avoid possible multiplication by None

2023-09-23 15:08:55 -07:00
parent d855f63985
commit a3a083c125
1 changed files with 5 additions and 1 deletions
@@ -71,7 +71,11 @@ class RichTqdmProgressAdapter:
        )
        self.unit_scale = unit_scale
        self.progress_bar = self.progress.add_task(
-            desc, total=total * self.unit_scale, unit=unit
+            desc,
+            total=total * self.unit_scale
+            if total is not None and self.unit_scale is not None
+            else None,
+            unit=unit,
        )

    def __enter__(self):