build: deprecate cpu and gpu extras (#155)

Author: shuangwu5

.github/workflows/run-tests-cpu.yml

@@ -30,7 +30,10 @@ jobs:
           python-version: '3.10'
 
       - name: Setup | dependencies
-        run: uv sync --frozen --extra cpu
+        run: |
+          uv sync --frozen --only-group dev --only-group docs
+          uv pip install --index-strategy unsafe-first-match torch==2.6.0+cpu torchvision==0.21.0+cpu . --extra-index-url https://download.pytorch.org/whl/cpu
+
       - name: Run tests
         run: uv run --no-sync pytest tests/
 

.github/workflows/run-tests-gpu.yml

@@ -37,7 +37,9 @@ jobs:
           python-version: '3.10'
 
       - name: Setup | dependencies
-        run: uv sync --frozen --extra gpu
+        run: |
+          uv sync --frozen --only-group dev --only-group docs
+          uv pip install .
 
       - name: Setup | Check for available GPU-s
         run: nvidia-smi

pyproject.toml

@@ -42,15 +42,6 @@ dependencies = [
     "torch>=2.6.0",
 ]
 
-[project.optional-dependencies]
-gpu = [
-    "torch>=2.6.0",
-]
-cpu = [
-    "torch==2.6.0+cpu; sys_platform == 'linux'",
-    "torch>=2.6.0; sys_platform != 'linux'",
-]
-
 [project.urls]
 homepage = "https://github.com/mostly-ai/mostlyai-qa"
 repository = "https://github.com/mostly-ai/mostlyai-qa"
@@ -76,22 +67,6 @@ docs = [
 
 [tool.uv]
 default-groups = ["dev", "docs"]
-conflicts = [
-  [
-    { extra = "cpu" },
-    { extra = "gpu" },
-  ],
-]
-
-[[tool.uv.index]]
-name = "pytorch-cpu"
-url = "https://download.pytorch.org/whl/cpu"
-explicit = true
-
-[tool.uv.sources]
-torch = [
-  { index = "pytorch-cpu", extra = "cpu", marker = "sys_platform == 'linux'"},
-]
 
 [tool.hatch.build.targets.sdist]
 include = ["mostlyai/qa"]