Prepare Needle for runtime-only deployments, such as serving tool-calling requests, without requiring the full training and checkpoint-export stack.
Move training dependencies out of the base installation and expose them through a dedicated `train` extra. Preserve the GPU and Metal extras for accelerated configurations, while keeping the default runtime dependency set minimal.
Changes:
- Add the `train` optional dependency group.
- Add `requirements-train.txt` for requirements-file users.
- Update setup scripts, CI, and documentation to install or explain the training extra where needed.
- Add packaging tests verifying that training dependencies are excluded from the runtime installation and included in the training setup.
Signed-off-by: Nima Kalantar <nima.kalantar@focusrite.com>
- Removed GCS utilities and TPU management scripts from the codebase.
- Updated project name to "cactus-needle" and version to "2.0.0" in pyproject.toml.
- Simplified setup scripts for virtual environment creation and dependency installation.
- Removed unnecessary dependencies from requirements.txt and setup scripts.
- Enhanced compatibility with Python 3.9 and above.
- Updated README and documentation references accordingly.
jakmro