Fix pyproject.toml for CI (GPU deps moved to gpu group)

Author: sindchad

pyproject.toml

@@ -7,27 +7,22 @@ authors = [{ name = "Sindhuja Chaduvula", email = "sindhuja.chaduvula@vectorinst
 license = "MIT"
 requires-python = ">=3.10,<3.12"
 
+# -----------------------------------------------------
+# Main dependencies (CI-safe, CPU-safe)
+# -----------------------------------------------------
 dependencies = [
-    # --- Your ML project dependencies ---
     "python-dotenv==1.1",
-    "torch==2.6.0+cu124",
-    "torchaudio==2.6.0+cu124",
-    "torchvision==0.21.0+cu124",
     "cmake==4.1.0",
     "ninja>=1.13.0",
     "pyarrow==15.0.2",
-    "unsloth[colab-new]",
     "einops>=0.8.1",
     "setuptools>=80.9.0",
     "wheel>=0.45.1",
     "build>=1.3.0",
     "trl==0.22.2",
     "peft>=0.17.1",
     "accelerate==1.10.1",
-    "bitsandbytes==0.48.1",
-    "unsloth-zoo>=2025.10.10",
     "wandb>=0.22.2",
-    "flash-attn==2.7.3",
     "sentence-transformers>=5.1.2",
     "openai>=2.6.1",
     "pydantic-settings>=2.11.0",
@@ -38,17 +33,25 @@ dependencies = [
     "opik>=1.9.33",
 ]
 
-[build-system]
-requires = ["hatchling"]
-build-backend = "hatchling.build"
-
-[tool.hatch.build.targets.wheel]
-packages = ["src/aixpert"]
-
-# ------------------------------------------------
-# Dependency groups (required by CI)
-# ------------------------------------------------
+# GPU-only dependencies MUST NOT be installed in CI
+# -----------------------------------------------------
+# These are installed only locally on GPU machines:
+#
+#   uv sync --group gpu
+#
 [dependency-groups]
+gpu = [
+    "torch==2.6.0+cu124",
+    "torchaudio==2.6.0+cu124",
+    "torchvision==0.21.0+cu124",
+    "bitsandbytes==0.48.1",
+    "flash-attn==2.7.3",
+    "unsloth[colab-new]",
+    "unsloth-zoo>=2025.10.10",
+]
+
+# Dev tools for linting, formatting, testing
+# -----------------------------------------------------
 dev = [
     "codecov>=2.1.13",
     "mypy>=1.14.1",
@@ -63,6 +66,8 @@ dev = [
     "types-pyyaml>=6.0.12.20250822",
 ]
 
+# Documentation dependencies (MkDocs)
+# -----------------------------------------------------
 docs = [
     "jinja2>=3.1.6",
     "mkdocs>=1.6.0",
@@ -73,22 +78,28 @@ docs = [
     "ipython>=8.31.0",
 ]
 
-# ------------------------------------------------
-# UV configuration
-# ------------------------------------------------
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/aixpert"]
+
+# -----------------------------------------------------
+# UV configuration – CI will install only dev + docs
+# -----------------------------------------------------
 [tool.uv]
+default-groups = ["dev", "docs"]
 index-url = "https://pypi.org/simple"
 extra-index-url = ["https://download.pytorch.org/whl/cu124"]
 index-strategy = "unsafe-best-match"
-default-groups = ["dev", "docs"]
 
 [tool.uv.sources]
 unsloth = { git = "https://github.com/unslothai/unsloth.git" }
 
-# ------------------------------------------------
-# Linting / type checking (required for CI)
-# ------------------------------------------------
-
+# -----------------------------------------------------
+# Mypy (Static Type Checking)
+# -----------------------------------------------------
 [tool.mypy]
 follow_imports = "normal"
 ignore_missing_imports = true
@@ -104,6 +115,9 @@ strict_equality = true
 extra_checks = true
 mypy_path = "src"
 
+# -----------------------------------------------------
+# Ruff (Linting & Formatting)
+# -----------------------------------------------------
 [tool.ruff]
 include = ["*.py", "pyproject.toml", "*.ipynb"]
 exclude = []
@@ -141,13 +155,11 @@ convention = "numpy"
 [tool.ruff.lint.pycodestyle]
 max-doc-length = 88
 
-# ------------------------------------------------
-# Pytest & Coverage (CI requirements)
-# ------------------------------------------------
+# -----------------------------------------------------
+# Pytest & Coverage
+# -----------------------------------------------------
 [tool.pytest.ini_options]
-markers = [
-    "integration_test: marks tests as integration tests",
-]
+markers = ["integration_test: marks tests as integration tests"]
 
 [tool.coverage.run]
 source = ["src/aixpert"]