AQIT 0.1.0
Loading...
Searching...
No Matches
aquin.sae_cli Namespace Reference

Functions

str|None _parse_flag (list[str] args, str name)
bool _has_flag (list[str] args, str name)
str _resolve_sae_model (str model_id)
None _ensure_compute_env ()
dict|None _build_sae_card (str tool_name, dict payload, dict|None sync_args=None)
None _sync_sae_result (dict ctx, str tool_name, dict args, dict payload)
None _print_sae_diff (dict payload)
str _require_session_model (str|None model_id)
str _resolve_train_model_id (str|None model_id, str|None activations)
None cmd_sae_diff (list[str] args)
list[int] _parse_train_layers (str spec, int n_layers)
str _resolve_train_name (str|None name, int layer, *, bool multi, str|None checkpoint=None)
None _train_one_layer (*, str mid, int layer, str tag, dict ctx, str|None checkpoint, str|None corpus, str|None activations, bool quick, bool balance, str|None balance_group, str|None save_path, int|None max_steps=None, int|None max_epochs=None)
None cmd_sae_train (list[str] args)
Path _resolve_user_sae_path (str model_id, str run_name, int layer)
None cmd_sae_catalog_metrics (list[str] args)
Path|None _default_user_sae_checkpoint ()
bool _is_dictionary_sae_path (Path path)
Path|None _resolve_align_sae_path (str|None raw)
None cmd_sae_align (list[str] args)
None cmd_sae (list[str] args)

Function Documentation

◆ _build_sae_card()

dict | None _build_sae_card ( str tool_name,
dict payload,
dict | None sync_args = None )
protected

Definition at line 42 of file sae_cli.py.

Referenced by _sync_sae_result().

◆ _default_user_sae_checkpoint()

Path | None _default_user_sae_checkpoint ( )
protected

Definition at line 527 of file sae_cli.py.

Referenced by _resolve_align_sae_path().

◆ _ensure_compute_env()

None _ensure_compute_env ( )
protected 
Match cmd_tool: shared model + local HTTP server for web popovers.

Definition at line 33 of file sae_cli.py.

Referenced by cmd_sae_align(), cmd_sae_catalog_metrics(), cmd_sae_diff(), and cmd_sae_train().

◆ _has_flag()

bool _has_flag ( list[str] args,
str name )
protected

Definition at line 22 of file sae_cli.py.

Referenced by cmd_sae_catalog_metrics(), and cmd_sae_train().

◆ _is_dictionary_sae_path()

bool _is_dictionary_sae_path ( Path path)
protected

Definition at line 539 of file sae_cli.py.

Referenced by _resolve_align_sae_path().

◆ _parse_flag()

str | None _parse_flag ( list[str] args,
str name )
protected

Definition at line 15 of file sae_cli.py.

Referenced by cmd_sae_align(), cmd_sae_catalog_metrics(), cmd_sae_diff(), and cmd_sae_train().

◆ _parse_train_layers()

list[int] _parse_train_layers ( str spec,
int n_layers )
protected

Definition at line 231 of file sae_cli.py.

Referenced by cmd_sae_catalog_metrics(), and cmd_sae_train().

◆ _print_sae_diff()

None _print_sae_diff ( dict payload)
protected

Definition at line 89 of file sae_cli.py.

Referenced by cmd_sae_diff().

◆ _require_session_model()

str _require_session_model ( str | None model_id)
protected

Definition at line 104 of file sae_cli.py.

Referenced by _resolve_train_model_id(), cmd_sae_catalog_metrics(), and cmd_sae_diff().

◆ _resolve_align_sae_path()

Path | None _resolve_align_sae_path ( str | None raw)
protected

Definition at line 554 of file sae_cli.py.

References _default_user_sae_checkpoint(), and _is_dictionary_sae_path().

Referenced by cmd_sae_align().

◆ _resolve_sae_model()

str _resolve_sae_model ( str model_id)
protected 
Return canonical model id.

Definition at line 26 of file sae_cli.py.

Referenced by cmd_sae_diff(), and cmd_sae_train().

◆ _resolve_train_model_id()

str _resolve_train_model_id ( str | None model_id,
str | None activations )
protected 
Session model, explicit --model, or capture-dir manifest (activations-only).

Definition at line 126 of file sae_cli.py.

References _require_session_model().

Referenced by cmd_sae_train().

◆ _resolve_train_name()

str _resolve_train_name ( str | None name,
int layer,
* ,
bool multi,
str | None checkpoint = None )
protected

Definition at line 249 of file sae_cli.py.

Referenced by cmd_sae_train().

◆ _resolve_user_sae_path()

Path _resolve_user_sae_path ( str model_id,
str run_name,
int layer )
protected

Definition at line 439 of file sae_cli.py.

Referenced by cmd_sae_catalog_metrics().

◆ _sync_sae_result()

None _sync_sae_result ( dict ctx,
str tool_name,
dict args,
dict payload )
protected

Definition at line 77 of file sae_cli.py.

References _build_sae_card().

Referenced by _train_one_layer(), cmd_sae_align(), and cmd_sae_diff().

◆ _train_one_layer()

None _train_one_layer ( * ,
str mid,
int layer,
str tag,
dict ctx,
str | None checkpoint,
str | None corpus,
str | None activations,
bool quick,
bool balance,
str | None balance_group,
str | None save_path,
int | None max_steps = None,
int | None max_epochs = None )
protected

Definition at line 268 of file sae_cli.py.

References _sync_sae_result().

Referenced by cmd_sae_train().

◆ cmd_sae()

None cmd_sae ( list[str] args)

Definition at line 625 of file sae_cli.py.

References cmd_sae_align(), cmd_sae_catalog_metrics(), and cmd_sae_train().

◆ cmd_sae_align()

None cmd_sae_align ( list[str] args)

Definition at line 570 of file sae_cli.py.

References _ensure_compute_env(), _parse_flag(), _resolve_align_sae_path(), and _sync_sae_result().

Referenced by cmd_sae().

◆ cmd_sae_catalog_metrics()

None cmd_sae_catalog_metrics ( list[str] args)

Definition at line 450 of file sae_cli.py.

References _ensure_compute_env(), _has_flag(), _parse_flag(), _parse_train_layers(), _require_session_model(), and _resolve_user_sae_path().

Referenced by cmd_sae().

◆ cmd_sae_diff()

None cmd_sae_diff ( list[str] args)

Definition at line 163 of file sae_cli.py.

References _ensure_compute_env(), _parse_flag(), _print_sae_diff(), _require_session_model(), _resolve_sae_model(), and _sync_sae_result().

◆ cmd_sae_train()

None cmd_sae_train ( list[str] args)

Definition at line 324 of file sae_cli.py.

References _ensure_compute_env(), _has_flag(), _parse_flag(), _parse_train_layers(), _resolve_sae_model(), _resolve_train_model_id(), _resolve_train_name(), and _train_one_layer().

Referenced by cmd_sae().