ml_switcheroo.cli.handlers.audit
================================

.. py:module:: ml_switcheroo.cli.handlers.audit

.. autoapi-nested-parse::

   Audit Command Handler.



Functions
---------

.. autoapisummary::

   ml_switcheroo.cli.handlers.audit.resolve_roots
   ml_switcheroo.cli.handlers.audit.handle_audit


Module Contents
---------------

.. py:function:: 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'.


.. py:function:: handle_audit(path: pathlib.Path, source_frameworks: List[str]) -> int

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

   :param path: Input source.
   :param source_frameworks: List of framework keys to scan for.