handle progress corner case

supervisely-ecosystem · Dec 13, 2024 · 3eafaef · 3eafaef
1 parent 3782da6
commit 3eafaef
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/train/src/ui/monitoring.py b/train/src/ui/monitoring.py
@@ -27,8 +27,12 @@
 
 def external_callback(progress: sly.tqdm_sly):
     percent = math.floor(progress.n / progress.total * 100)
-    fields = [
-        {"field": f"data.progressBenchmark", "payload": progress.desc},
+    fields = []
+    if hasattr(progress, "desc"):
+        fields.append({"field": f"data.progressBenchmark", "payload": progress.desc})
+    elif hasattr(progress, "message"):
+        fields.append({"field": f"data.progressBenchmark", "payload": progress.message})
+    fields += [
         {"field": f"data.progressCurrentBenchmark", "payload": progress.n},
         {"field": f"data.progressTotalBenchmark", "payload": progress.total},
         {"field": f"data.progressPercentBenchmark", "payload": percent},