no extra

pyproject.toml

@@ -28,7 +28,6 @@ dependencies = [
     "setuptools>=75.0.0,<76",
     "numpy>=1.26.3,<2",
     "pandas~=2.2.0",
-    "torch>=2.5.1",
     "pyarrow>=16.0.0",
     "joblib>=1.3.0,<2",
     "psutil>=5.9.5,<6",  # upgrade when colab psutil is updated
@@ -41,24 +40,12 @@ dependencies = [
     "formatron>=0.4.11,<0.5",
     "json-repair>=0.30.0,<0.31",
     "rich>=13.9.4,<14",
-]
-
-[project.optional-dependencies]
-gpu = [
-    "torch>=2.5.1",
-    "bitsandbytes>=0.43.3,<0.44",
-    "vllm>=0.6.4,<0.7",
-]
-rocm = [
     "pytorch-triton-rocm>=3.1.0",
     "torch>=2.5.1",
     "torchvision>=0.20.1",
     # "bitsandbytes>=0.43.3,<0.44",  # need to be built from source
     # "vllm>=0.6.4,<0.7",  # need to be built from source
 ]
-cpu = [
-    "torch>=2.5.1"
-]
 
 [dependency-groups]
 test = ["pytest>=7.1.2,<8"]
@@ -85,18 +72,6 @@ documentation = "https://mostly-ai.github.io/mostlyai-engine/"
 
 [tool.uv]
 default-groups = ["test", "dev", "docs"]
-conflicts = [
-  [
-    { extra = "cpu" },
-    { extra = "gpu" },
-    { extra = "rocm" },
-  ],
-]
-
-[[tool.uv.index]]
-name = "pytorch-cpu"
-url = "https://download.pytorch.org/whl/cpu"
-explicit = true
 
 [[tool.uv.index]]
 name = "pytorch-rocm"
@@ -105,14 +80,13 @@ explicit = true
 
 [tool.uv.sources]
 torch = [
-  { index = "pytorch-cpu", extra = "cpu" },
-  { index = "pytorch-rocm", extra = "rocm" },
+  { index = "pytorch-rocm" },
 ]
 torchvision = [
-  { index = "pytorch-rocm", extra = "rocm" },
+  { index = "pytorch-rocm" },
 ]
 pytorch-triton-rocm = [
-  { index = "pytorch-rocm", extra = "rocm" },
+  { index = "pytorch-rocm" },
 ]