[CI/Docs] Improve aarch64/DGX Spark support for dev setup (#38057)
Signed-off-by: Ben Browning <bbrownin@redhat.com> Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com> Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> Co-authored-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
This commit is contained in:
28
AGENTS.md
28
AGENTS.md
@@ -39,6 +39,8 @@ If work is duplicate/trivial busywork, **do not proceed**. Return a short explan
|
|||||||
|
|
||||||
## 2. Development Workflow
|
## 2. Development Workflow
|
||||||
|
|
||||||
|
- **Never use system `python3` or bare `pip`/`pip install`.** All Python commands must go through `uv` and `.venv/bin/python`.
|
||||||
|
|
||||||
### Environment setup
|
### Environment setup
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
@@ -58,33 +60,33 @@ pre-commit install
|
|||||||
|
|
||||||
```bash
|
```bash
|
||||||
# If you are only making Python changes:
|
# If you are only making Python changes:
|
||||||
VLLM_USE_PRECOMPILED=1 uv pip install -e .
|
VLLM_USE_PRECOMPILED=1 uv pip install -e . --torch-backend=auto
|
||||||
|
|
||||||
# If you are also making C/C++ changes:
|
# If you are also making C/C++ changes:
|
||||||
uv pip install -e .
|
uv pip install -e . --torch-backend=auto
|
||||||
```
|
```
|
||||||
|
|
||||||
### Running tests
|
### Running tests
|
||||||
|
|
||||||
Tests require extra dependencies.
|
> Requires [Environment setup](#environment-setup) and [Installing dependencies](#installing-dependencies).
|
||||||
All versions for test dependencies should be read from `requirements/test.txt`
|
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
# Install bare minimum test dependencies:
|
# Install test dependencies.
|
||||||
uv pip install pytest pytest-asyncio tblib
|
# requirements/test.txt is pinned to x86_64; on other platforms, use the
|
||||||
|
# unpinned source file instead:
|
||||||
# Install additional test dependencies as needed, or install them all as follows:
|
uv pip install -r requirements/test.in # resolves for current platform
|
||||||
|
# Or on x86_64:
|
||||||
uv pip install -r requirements/test.txt
|
uv pip install -r requirements/test.txt
|
||||||
|
|
||||||
# Run specific test from specific test file
|
# Run a specific test file (use .venv/bin/python directly;
|
||||||
pytest tests/path/to/test.py -v -s -k test_name
|
# `source activate` does not persist in non-interactive shells):
|
||||||
|
.venv/bin/python -m pytest tests/path/to/test_file.py -v
|
||||||
# Run all tests in directory
|
|
||||||
pytest tests/path/to/dir -v -s
|
|
||||||
```
|
```
|
||||||
|
|
||||||
### Running linters
|
### Running linters
|
||||||
|
|
||||||
|
> Requires [Environment setup](#environment-setup).
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
# Run all pre-commit hooks on staged files:
|
# Run all pre-commit hooks on staged files:
|
||||||
pre-commit run
|
pre-commit run
|
||||||
|
|||||||
@@ -61,7 +61,7 @@ runai-model-streamer[s3,gcs,azure]==0.15.7
|
|||||||
fastsafetensors>=0.2.2 # 0.2.2 contains important fixes for multi-GPU mem usage
|
fastsafetensors>=0.2.2 # 0.2.2 contains important fixes for multi-GPU mem usage
|
||||||
instanttensor>=0.1.5
|
instanttensor>=0.1.5
|
||||||
pydantic>=2.12 # 2.11 leads to error on python 3.13
|
pydantic>=2.12 # 2.11 leads to error on python 3.13
|
||||||
decord==0.6.0
|
decord==0.6.0; platform_machine == "x86_64"
|
||||||
terratorch >= 1.2.2 # Required for Prithvi tests
|
terratorch >= 1.2.2 # Required for Prithvi tests
|
||||||
imagehash # Required for Prithvi tests
|
imagehash # Required for Prithvi tests
|
||||||
segmentation-models-pytorch > 0.4.0 # Required for Prithvi tests
|
segmentation-models-pytorch > 0.4.0 # Required for Prithvi tests
|
||||||
|
|||||||
Reference in New Issue
Block a user