From da680901eb33bcfbabe76f99632c2b4a17082183 Mon Sep 17 00:00:00 2001
From: wassname <1103714+wassname@users.noreply.github.com>
Date: Sun, 12 Jul 2026 10:07:07 +0800
Subject: [PATCH] demo: fix demo_steer return annotation list->dict (stale
 after {summary,detail} change)

Co-Authored-By: Claudypoo <288921227+claudypoo@users.noreply.github.com>
---
 jsteer/demo.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/jsteer/demo.py b/jsteer/demo.py
index 60842d0..2811ae8 100644
--- a/jsteer/demo.py
+++ b/jsteer/demo.py
@@ -368,7 +368,7 @@ def show_steer(jac: Jacobian, model, tok, vec, user_msg: str, *,
 @torch.no_grad()
 def demo_steer(jac: Jacobian, model, tok, vecs: dict, user_msg: str, *,
                rubric: str | None = None, readout: dict = DIGIT, budget: int = 6,
-               max_new_tokens: int = 256, seed: int = 0) -> list[dict]:
+               max_new_tokens: int = 256, seed: int = 0) -> dict:
     """THE steering demo, one call for every notebook. For each named vector: a clear
     heading, a search for the strongest coherent steer each way, the raw generations at
     [-C*, -C*/2, 0, +C*/2, +C*] (comparable qualitative outputs, same prompt), then ONE