injector_fw

LibCST Transformer for Injecting Framework Mappings.

This package provides the logic to modify framework adapter files (e.g. torch.py) by locating the specific class registered for a framework and injecting a new StandardMap definition into its definitions property.

It handles:

1. Definitions Injection: Appending the mapping to the definitions dictionary.

2. Smart Import Injection: Analyzing the target API path (e.g. scipy.special.erf) and injecting necessary top-level imports (import scipy) if missing.

3. Variant Parameter Injection: Supporting inject_args for adding fixed arguments.

4. Complex Literal Support: Recursively converting Lists, Tuples, and Dicts to CST nodes.

Submodules

• injector_fw.core
• injector_fw.utils