Make feature tables sortable

scripts/explorer_app.py

@@ -123,13 +123,6 @@ def _get_clip():
     return _clip_handle[0]
 
 
-def _center_umap(coords: np.ndarray) -> np.ndarray:
-    """Shift UMAP coordinates so the mean of live (non-NaN) points is (0, 0)."""
-    live = ~np.isnan(coords[:, 0])
-    coords[live] -= coords[live].mean(axis=0)
-    return coords
-
-
 # ---------- Load all datasets into a unified list ----------
 
 def _load_dataset_dict(path, label, sae_url=None):
@@ -169,8 +162,8 @@ def _load_dataset_dict(path, label, sae_url=None):
         'feature_frequency': d['feature_frequency'],
         'feature_mean_act':  d['feature_mean_act'],
         'feature_p75_val':   d['feature_p75_val'],
-        'umap_coords':       _center_umap(d['umap_coords'].numpy()),
-        'dict_umap_coords':  _center_umap(d['dict_umap_coords'].numpy()),
+        'umap_coords':       d['umap_coords'].numpy(),
+        'dict_umap_coords':  d['dict_umap_coords'].numpy(),
         'clip_scores':       cs,
         'clip_vocab':        d.get('clip_text_vocab', None),
         'clip_embeds':       d.get('clip_feature_embeds', None),
@@ -1215,7 +1208,7 @@ feature_table = DataTable(
                     formatter=NumberFormatter(format="0.0000")),
         TableColumn(field="name", title="Name", width=200),
     ],
-    width=500, height=500, sortable=False, index_position=None,
+    width=500, height=500, sortable=True, index_position=None,
 )
 
 # Search state: None = no filter, otherwise a set of matching feature indices
@@ -1611,7 +1604,7 @@ patch_feat_table = DataTable(
         TableColumn(field="mean_act",    title="Mean Act",  width=80,
                     formatter=NumberFormatter(format="0.0000")),
     ],
-    width=310, height=350, index_position=None, sortable=False, visible=False,
+    width=310, height=350, index_position=None, sortable=True, visible=False,
 )
 patch_info_div = Div(
     text="<i>Load an image, then click patches to find top features.</i>",
@@ -1780,7 +1773,7 @@ if HAS_CLIP:
                         formatter=NumberFormatter(format="0.0000")),
             TableColumn(field="name",        title="Name", width=160),
         ],
-        width=470, height=300, index_position=None, sortable=False,
+        width=470, height=300, index_position=None, sortable=True,
     )
 
     def _do_clip_search():