ml_switcheroo.cli.handlers.learning
===================================

.. py:module:: ml_switcheroo.cli.handlers.learning

.. autoapi-nested-parse::

   Interactive and Learning Command Handlers.

   This module provides tools for:
   1.  **Wizard**: Interactive CLI to categorize unmapped APIs.
   2.  **Harvest**: Learning mappings from manual test files.



Functions
---------

.. autoapisummary::

   ml_switcheroo.cli.handlers.learning.handle_wizard
   ml_switcheroo.cli.handlers.learning.handle_harvest


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

.. py:function:: handle_wizard(package: str) -> int

   Handles the 'wizard' command for interactive mapping discovery.

   :param package: The name of the python package to inspect (e.g., 'torch').

   :returns: Exit code (0 for success).
   :rtype: int


.. py:function:: handle_harvest(path: pathlib.Path, target: str, dry_run: bool) -> int

   Handles the 'harvest' command to learn mappings from manual tests.

   :param path: File or directory containing python test files.
   :param target: The framework target used in the tests (e.g., 'jax').
   :param dry_run: If True, prints changes without writing to disk.

   :returns: Exit code (0 for success, 1 for path errors).
   :rtype: int