jashshah999 commited on
Commit
423c26c
·
1 Parent(s): be76781

fix warn count to show individual messages, bump to c5ec8b2 (5% frozen threshold)

Browse files
Files changed (2) hide show
  1. app.py +6 -4
  2. requirements.txt +1 -1
app.py CHANGED
@@ -241,14 +241,16 @@ def _format_overall_banner(report: DiagnosticReport) -> str:
241
 
242
 
243
  def _format_summary_cards(report: DiagnosticReport) -> str:
244
- counts = report.summary_counts
 
 
245
  cards = [
246
  ("Episodes", f"{report.total_episodes:,}" if report.total_episodes is not None else "--"),
247
  ("Frames", f"{report.total_frames:,}" if report.total_frames is not None else "--"),
248
  ("FPS", str(report.fps) if report.fps is not None else "--"),
249
- ("Pass", str(counts["PASS"])),
250
- ("Warn", str(counts["WARN"])),
251
- ("Fail", str(counts["FAIL"])),
252
  ]
253
  cards_html = "".join(
254
  f'<div class="summary-card"><div class="label">{label}</div>'
 
241
 
242
 
243
  def _format_summary_cards(report: DiagnosticReport) -> str:
244
+ n_pass = sum(1 for r in report.results if r.severity == Severity.PASS)
245
+ n_warn = sum(1 for r in report.results for m in r.messages if m.severity == Severity.WARN)
246
+ n_fail = sum(1 for r in report.results for m in r.messages if m.severity == Severity.FAIL)
247
  cards = [
248
  ("Episodes", f"{report.total_episodes:,}" if report.total_episodes is not None else "--"),
249
  ("Frames", f"{report.total_frames:,}" if report.total_frames is not None else "--"),
250
  ("FPS", str(report.fps) if report.fps is not None else "--"),
251
+ ("Pass", str(n_pass)),
252
+ ("Warn", str(n_warn)),
253
+ ("Fail", str(n_fail)),
254
  ]
255
  cards_html = "".join(
256
  f'<div class="summary-card"><div class="label">{label}</div>'
requirements.txt CHANGED
@@ -1,3 +1,3 @@
1
  gradio>=5.0
2
- lerobot-doctor @ git+https://github.com/jashshah999/lerobot-doctor.git@dca9752
3
  huggingface-hub
 
1
  gradio>=5.0
2
+ lerobot-doctor @ git+https://github.com/jashshah999/lerobot-doctor.git@c5ec8b2
3
  huggingface-hub