mirror dev deps

.github/workflows/tests.yml

@@ -82,7 +82,7 @@ jobs:
       - name: Install dependencies
         run: |
           uv pip show --system torch
-          uv pip install --system wheel pytest pytest-xdist pytest-cov
+          uv pip install --system wheel
           printf "torch==${{ matrix.pytorch_version }}\n" > torch-constraints.txt
           uv pip install --system --no-build-isolation -e ".[dev]" --constraints torch-constraints.txt
           set -o pipefail
@@ -155,7 +155,7 @@ jobs:
       - name: Install dependencies
         run: |
           uv pip show --system torch
-          uv pip install --system wheel pytest pytest-xdist pytest-cov
+          uv pip install --system wheel
           uv pip install --system build
           python -m build --sdist
           uv pip install --system dist/*.tar.gz

pyproject.toml

@@ -107,6 +107,24 @@ optimizers = [
 ray = ["ray[train]"]
 vllm = ["vllm>=0.10.0"]
 llmcompressor = ["llmcompressor>=0.5.1"]
+dev = [
+    "pytest",
+    "pytest-cov",
+    "pytest-retry",
+    "pytest-sugar",
+    "pytest-xdist",
+    "codecov",
+    "codecov-cli",
+    "tbparse",
+    "ruff",
+    "mypy",
+    "pre-commit",
+    "types-requests",
+    "quartodoc",
+    "jupyter",
+    "blobfile",
+    "tiktoken",
+]
 
 [project.scripts]
 axolotl = "axolotl.cli.main:main"
@@ -173,6 +191,7 @@ python_files = ["test_*.py", "*_test.py"]
 addopts = "-v --tb=short"
 
 # UV specific configuration
+[tool.uv]
 prerelease = "allow"
 default-groups = ["default"]
 conflicts = [