aquin.engine.tools.inspect Namespace Reference

Functions
int	_resolve_inspection_sae_layer (str model_id, Any layer_raw, *, dict[str, Any]|None memory=None, str command="trace")
dict|None	_build_color_map (list[dict] attribution)
list[dict]	_build_trace_results (list[dict] logit_lens)
dict[str, Any]	run_full_inspection (dict[str, Any] args, dict[str, Any] ctx)
dict[str, Any]	get_feature_logits (dict[str, Any] args, dict[str, Any] ctx)
dict[str, Any]	get_feature_neighbors (dict[str, Any] args, dict[str, Any] ctx)

Function Documentation

_build_color_map()

dict | None _build_color_map ( list[dict] attribution )

protected

Definition at line 35 of file inspect.py.

Referenced by run_full_inspection().

_build_trace_results()

list[dict] _build_trace_results ( list[dict] logit_lens )

protected

Definition at line 60 of file inspect.py.

Referenced by run_full_inspection().

_resolve_inspection_sae_layer()

int _resolve_inspection_sae_layer ( str model_id, Any layer_raw, * , dict[str, Any] | None memory = None, str command = "trace" )

protected

Resolve SAE layer from explicit arg or session memory; never silently default.

Definition at line 17 of file inspect.py.

Referenced by get_feature_logits(), get_feature_neighbors(), and run_full_inspection().

get_feature_logits()

dict[str, Any] get_feature_logits	(	dict[str, Any]	args,
		dict[str, Any]	ctx )

Definition at line 262 of file inspect.py.

References _resolve_inspection_sae_layer().

get_feature_neighbors()

dict[str, Any] get_feature_neighbors	(	dict[str, Any]	args,
		dict[str, Any]	ctx )

Definition at line 353 of file inspect.py.

References _resolve_inspection_sae_layer().

run_full_inspection()

dict[str, Any] run_full_inspection	(	dict[str, Any]	args,
		dict[str, Any]	ctx )

Definition at line 106 of file inspect.py.

References _build_color_map(), _build_trace_results(), and _resolve_inspection_sae_layer().