Fix package buttons not displaying in relevant stages

mlcommons · Jul 31, 2024 · 35a09f0 · 35a09f0
1 parent fac2940
commit 35a09f0
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/scripts/monitor/rano_monitor/widgets/summary.py b/scripts/monitor/rano_monitor/widgets/summary.py
@@ -1,6 +1,6 @@
 import os
 import pandas as pd
-from rano_monitor.constants import REVIEW_FILENAME, REVIEWED_FILENAME
+from rano_monitor.constants import REVIEW_FILENAME, REVIEWED_FILENAME, MANUAL_REVIEW_STAGE, DONE_STAGE
 from rano_monitor.messages import InvalidSubjectsUpdated
 from rano_monitor.messages import ReportUpdated
 from rano_monitor.messages import AnnotationsLoaded
@@ -66,7 +66,10 @@ def update_summary(self):
             # Attach
             status_percents["DONE"] = 0.0
 
-        package_btns.display = "MANUAL_REVIEW_REQUIRED" in status_percents
+        abs_status = display_report_df["status"].abs()
+        is_beyond_manual_review = (abs_status >= MANUAL_REVIEW_STAGE) 
+        is_not_done = (abs_status < DONE_STAGE)
+        package_btns.display = any(is_beyond_manual_review & is_not_done)
 
         widgets = []
         for name, val in status_percents.items():