Limit static synthesizer output length.

sandialabs · Oct 30, 2023 · 000d7eb · 000d7eb
1 parent 23cf42f
commit 000d7eb
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/riid/data/synthetic/__init__.py b/riid/data/synthetic/__init__.py
@@ -60,8 +60,11 @@ def _report_progress(self, n_samples_expected, batch_name):
         percent_complete = 100 * self._n_samples_synthesized / n_samples_expected
         msg = (
             f"Synthesizing ... {percent_complete:.0f}% "
-            f"(currently on {batch_name})"
+            f"(currently on {batch_name}"
         )
+        MAX_MSG_LEN = 80
+        msg = (msg[:MAX_MSG_LEN] + "...") if len(msg) > MAX_MSG_LEN else msg
+        msg += ")"
         print("\033[K" + msg, end="\r")
 
     def _report_completion(self, delay):