ml_switcheroo.cli.handlers.audit

Audit Command Handler.

Functions

resolve_roots(→ Set[str])	Expands framework keys (e.g. 'flax_nnx') to python module roots (e.g. 'flax', 'jax').
handle_audit(→ int)	Scans a directory/file to determine coverage against the Knowledge Base.

Module Contents

ml_switcheroo.cli.handlers.audit.resolve_roots(framework_keys: List[str]) → Set[str]

Expands framework keys (e.g. ‘flax_nnx’) to python module roots (e.g. ‘flax’, ‘jax’). This ensures that querying for ‘flax_nnx’ correctly finds ‘import flax’.

ml_switcheroo.cli.handlers.audit.handle_audit(path: pathlib.Path, source_frameworks: List[str]) → int

Scans a directory/file to determine coverage against the Knowledge Base.

Parameters:

• path – Input source.

• source_frameworks – List of framework keys to scan for.