Type Hints for ML Code

npx skills add majiayu000/claude-skill-registry --skill ml-type-hints-apassuello-multimodal-insight-e --agent claude-code

Same command for any agent — swap --agent for codex, cursor, copilot.

# Type Hints for ML Code - Constitutional AI ## When to Apply Automatically activate when: - Adding type hints to ML code - Addressing mypy type errors - Working with PyTorch tensors, models, or optimizers - Dealing with HuggingFace transformers types ## Project Type Checking Status **Current mypy status:** - **41 errors remaining** (documented in MYPY_ANALYSIS_REPORT.md) - **Status**: Accepted as reasonable for ML research code - **CI behavior**: `continue-on-error: true` (doesn't fail builds) **Key insight**: Perfect type coverage is not the goal for ML code. Prioritize correctness and readability over type perfection. ## When to Use Type Hints ### ✅ Do Use Type Hints For 1. **Public API functions** ```python def load_model(model_name: str) -> tuple[AutoModelForCausalLM, AutoTokenizer]: """Load model and tokenizer.""" pass ``` 2. **Configuration dataclasses** ```python @dataclass class TrainingConfig: learning_rate: float batch_size: int num_epochs: int ``` 3. **Clear input/output types** ```python def evaluate_text(text: str, framework: ConstitutionalFramework) -> dict[str, Any]: """Evaluate text against principles.""" pass ``` 4. **Helper functions with simple types** ```python

Run mypy on all source code
mypy constitutional_ai/ --ignore-missing-imports
Run on specific module
mypy constitutional_ai/framework.py
Show error codes (useful for targeted ignores)
mypy constitutional_ai/ --show-error-codes
Strict mode (educational, will fail)
mypy constitutional_ai/ --strict