SYMBOL INDEX (57 symbols across 9 files)

FILE: data.py
  function load_gsm8k (line 8) | def load_gsm8k(split: str = "test", cache_dir: Optional[str] = None) -> ...
  function load_aime2025 (line 21) | def load_aime2025(split: str = "train", cache_dir: Optional[str] = None)...
  function load_aime2024 (line 34) | def load_aime2024(split: str = "train", cache_dir: Optional[str] = None)...
  function load_gpqa_diamond (line 47) | def load_gpqa_diamond(split: str = "test", cache_dir: Optional[str] = No...
  function load_arc_easy (line 60) | def load_arc_easy(split: str = "test", cache_dir: Optional[str] = None) ...
  function load_arc_challenge (line 97) | def load_arc_challenge(split: str = "test", cache_dir: Optional[str] = N...
  function load_winogrande (line 132) | def load_winogrande(
  function load_mbppplus (line 153) | def load_mbppplus(
  function load_humanevalplus (line 177) | def load_humanevalplus(
  function load_medqa (line 202) | def load_medqa(split=None, subset=None, cache_dir=None):

FILE: methods/__init__.py
  class Agent (line 6) | class Agent:
  function default_agents (line 11) | def default_agents() -> List[Agent]:

FILE: methods/baseline.py
  class BaselineMethod (line 8) | class BaselineMethod:
    method __init__ (line 9) | def __init__(
    method run_batch (line 30) | def run_batch(self, items: List[Dict]) -> List[Dict]:
    method run_item (line 119) | def run_item(self, item: Dict) -> Dict:

FILE: methods/latent_mas.py
  class LatentMASMethod (line 17) | class LatentMASMethod:
    method __init__ (line 18) | def __init__(
    method _slice_tensor (line 54) | def _slice_tensor(tensor: torch.Tensor, tokens_to_keep: int) -> torch....
    method _truncate_past (line 61) | def _truncate_past(self, past_kv: Optional[Tuple], tokens_to_keep: int...
    method run_batch (line 82) | def run_batch(self, items: List[Dict]) -> List[Dict]:
    method run_batch_vllm (line 251) | def run_batch_vllm(self, items: List[Dict]) -> List[Dict]:
    method run_item (line 440) | def run_item(self, item: Dict) -> Dict:

FILE: methods/text_mas.py
  class TextMASMethod (line 11) | class TextMASMethod:
    method __init__ (line 12) | def __init__(
    method run_batch (line 33) | def run_batch(self, items: List[Dict]) -> List[Dict]:
    method run_item (line 179) | def run_item(self, item: Dict) -> Dict:

FILE: models.py
  function _ensure_pad_token (line 15) | def _ensure_pad_token(tokenizer: AutoTokenizer) -> None:
  function _past_length (line 23) | def _past_length(past_key_values: Optional[Tuple]) -> int:
  class ModelWrapper (line 30) | class ModelWrapper:
    method __init__ (line 31) | def __init__(self, model_name: str, device: torch.device, use_vllm: bo...
    method render_chat (line 87) | def render_chat(self, messages: List[Dict], add_generation_prompt: boo...
    method prepare_chat_input (line 102) | def prepare_chat_input(
    method prepare_chat_batch (line 117) | def prepare_chat_batch(
    method vllm_generate_text_batch (line 139) | def vllm_generate_text_batch(
    method _build_latent_realign_matrix (line 158) | def _build_latent_realign_matrix(self, model, device, args) -> Tuple[t...
    method _ensure_latent_realign_matrix (line 187) | def _ensure_latent_realign_matrix(self, model, device, args) -> Tuple[...
    method _apply_latent_realignment (line 204) | def _apply_latent_realignment(self, hidden: torch.Tensor, model: torch...
    method generate_text_batch (line 216) | def generate_text_batch(
    method tokenize_text (line 269) | def tokenize_text(self, text: str) -> torch.Tensor:
    method generate_latent_batch (line 277) | def generate_latent_batch(
    method generate_latent_batch_hidden_state (line 353) | def generate_latent_batch_hidden_state(

FILE: prompts.py
  function build_agent_message_sequential_latent_mas (line 2) | def build_agent_message_sequential_latent_mas(role: str, question: str, ...
  function build_agent_message_hierarchical_latent_mas (line 118) | def build_agent_message_hierarchical_latent_mas(role: str, question: str...
  function build_agent_messages_sequential_text_mas (line 341) | def build_agent_messages_sequential_text_mas(role: str, question: str, c...
  function build_agent_messages_hierarchical_text_mas (line 511) | def build_agent_messages_hierarchical_text_mas(role: str, question: str,...
  function build_agent_messages_single_agent (line 694) | def build_agent_messages_single_agent(question: str, args=None):

FILE: run.py
  function evaluate (line 26) | def evaluate(preds: List[Dict]) -> Tuple[float, int]:
  function process_batch (line 33) | def process_batch(
  function main (line 84) | def main():

FILE: utils.py
  function set_seed (line 10) | def set_seed(seed: int) -> None:
  function auto_device (line 18) | def auto_device(device: Optional[str] = None) -> torch.device:
  function extract_gsm8k_answer (line 26) | def extract_gsm8k_answer(text: str) -> Optional[str]:
  function extract_gold (line 39) | def extract_gold(text: str) -> Optional[str]:
  function normalize_answer (line 44) | def normalize_answer(ans: Optional[str]) -> Optional[str]:
  function extract_markdown_python_block (line 50) | def extract_markdown_python_block(text: str) -> Optional[str]:
  function run_with_timeout (line 61) | def run_with_timeout(code, timeout):