ml_switcheroo.importers.array_api_reader
========================================

.. py:module:: ml_switcheroo.importers.array_api_reader

.. autoapi-nested-parse::

   Importer for Python Array API Standard Stubs.

   This module parses .py stub files from the official Array API standard repo,
   extracting function signatures, type hints, and docstrings to build the
   Semantic Knowledge Base.

   Feature 027 Update:
   Now extracts type hints (e.g. ``x: Array``, ``axis: int``) to support Better Fuzzing.



Classes
-------

.. autoapisummary::

   ml_switcheroo.importers.array_api_reader.ArrayApiSpecImporter


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

.. py:class:: ArrayApiSpecImporter

   Parses Python stub files (``*.py``) using the built-in ``ast`` module.


   .. py:method:: parse_folder(root_dir: pathlib.Path) -> Dict[str, Any]

      Parses Array API Python Stubs (``*.py``) in the target directory.

      :param root_dir: Path to the folder containing .py stubs (e.g. ``src/array_api_stubs/_2023_12``).

      :returns: Dict mapping function/constant names to their definitions.