Marlin Lee commited on
Commit
68497cf
·
1 Parent(s): 5891413

Sync explorer_app.py and clip_utils.py from main repo

Browse files
Files changed (1) hide show
  1. scripts/explorer_app.py +11 -3
scripts/explorer_app.py CHANGED
@@ -666,6 +666,14 @@ def make_cross_sae_comparison_html(ds_a, feat_a, ds_b, feat_b, n=4, size=160):
666
  items_a = _collect(ds_a, feat_a)
667
  items_b = _collect(ds_b, feat_b)
668
 
 
 
 
 
 
 
 
 
669
  def _grid_html(items, label, color):
670
  header = (f'<div style="background:{color};color:#fff;font-size:11px;font-weight:bold;'
671
  f'text-align:center;padding:4px;border-radius:4px;margin-bottom:6px">{label}</div>')
@@ -677,10 +685,10 @@ def make_cross_sae_comparison_html(ds_a, feat_a, ds_b, feat_b, n=4, size=160):
677
  f' style="border:1px solid #ccc;border-radius:3px;display:block"/>'
678
  f'<div style="font-size:9px;color:#555;margin-top:2px">{cap}</div></div>')
679
  grid += '</div>'
680
- return header + grid
681
 
682
- label_a = f"{ds_a['label']} — feat {feat_a}"
683
- label_b = f"{ds_b['label']} — feat {feat_b}"
684
  col_a = _grid_html(items_a, label_a, "#2563a8")
685
  col_b = _grid_html(items_b, label_b, "#b85c00")
686
 
 
666
  items_a = _collect(ds_a, feat_a)
667
  items_b = _collect(ds_b, feat_b)
668
 
669
+ def _strip_dim(label):
670
+ """Remove parenthetical dimension info like '(d=32K)' or '(d=32K, k=160)'."""
671
+ out = label
672
+ while '(' in out and ')' in out:
673
+ l, r = out.index('('), out.index(')')
674
+ out = out[:l].rstrip() + out[r+1:]
675
+ return out.strip(' —').strip()
676
+
677
  def _grid_html(items, label, color):
678
  header = (f'<div style="background:{color};color:#fff;font-size:11px;font-weight:bold;'
679
  f'text-align:center;padding:4px;border-radius:4px;margin-bottom:6px">{label}</div>')
 
685
  f' style="border:1px solid #ccc;border-radius:3px;display:block"/>'
686
  f'<div style="font-size:9px;color:#555;margin-top:2px">{cap}</div></div>')
687
  grid += '</div>'
688
+ return f'<div style="display:flex;flex-direction:column">{header}{grid}</div>'
689
 
690
+ label_a = f"{_strip_dim(ds_a['label'])} — feat {feat_a}"
691
+ label_b = f"{_strip_dim(ds_b['label'])} — feat {feat_b}"
692
  col_a = _grid_html(items_a, label_a, "#2563a8")
693
  col_b = _grid_html(items_b, label_b, "#b85c00")
694