activation_replay.py File Reference

Go to the source code of this file.

Namespaces
namespace	aquin
namespace	aquin.compute
namespace	aquin.compute.activation_replay

Functions
dict[str, Any]	aquin.compute.activation_replay._read_json (Path path)
dict[str, Any]	aquin.compute.activation_replay.load_capture_root (str|Path path)
dict[str, Any]	aquin.compute.activation_replay._flatten_probe_row (dict[str, Any] row, dict[str, dict[str, Any]] probe_lookup)
list[tuple[str, str]]	aquin.compute.activation_replay._parse_filter (str|None expr)
bool	aquin.compute.activation_replay._matches (dict[str, Any] row, list[tuple[str, str]] filters)
dict[str, Any]	aquin.compute.activation_replay.replay_capture (str|Path path, *, str|None filter_expr=None, str|None group_by=None, int|None limit=None)
dict[str, list[int]]	aquin.compute.activation_replay._probe_groups (list[dict[str, Any]] probes, str field)
bool	aquin.compute.activation_replay._usable_groups (dict[str, list[int]] grouped)
tuple[str, dict[str, list[int]], bool]	aquin.compute.activation_replay.resolve_compare_group (list[dict[str, Any]] probes, str requested)
None	aquin.compute.activation_replay._write_json (Path path, dict[str, Any] data)
dict[str, Any]	aquin.compute.activation_replay.encode_capture_sae_features (str|Path path, *, int|None sae_layer=None, Any|None sae=None)
dict[str, Any]	aquin.compute.activation_replay.compare_capture_features (str|Path path, *, str group, int top_k=10, bool encode_sae=True, int|None sae_layer=None, Any|None sae=None)

Variables
tuple	aquin.compute.activation_replay.COHORT_FIELDS = ("group", "label", "lang", "stressor", "source")