Contributing¶
We welcome contributions! This guide will help you get started.
Development Setup¶
-
Fork and clone the repository:
-
Create a virtual environment:
-
Install in development mode:
-
Install pre-commit hooks:
Code Style¶
We use the following tools for code quality:
- Black: Code formatting
- isort: Import sorting
- Flake8: Linting
Run before committing:
Testing¶
Run the test suite:
With coverage:
Pull Request Process¶
-
Create a feature branch:
-
Make your changes and commit:
-
Push to your fork:
-
Open a Pull Request
Adding a New Model¶
-
Create
models/your_model.py: -
Register in
models/__init__.py: -
Add tests in
tests/test_models.py -
Add documentation in
docs/models/
Adding a New Loss¶
-
Add to
models/losses.py: -
Register in
create_loss()factory -
Add tests in
tests/test_losses.py
Reporting Issues¶
Please use the issue templates: - Bug Report - Feature Request
Include: - Python version - PyTorch version - OS - Steps to reproduce - Expected vs actual behavior