cpu torch by default

Author: callum-jpg

pyproject.toml

@@ -23,67 +23,31 @@ dependencies = [
     "geopandas>=1.0",
     "rasterio>=1.4",
     "spatial_image>=1.2",
-    "spatialdata>=0.6.1,<0.7",
+    "spatialdata>=0.6.1",
     "ome-zarr>=0.13",
     "zarr>=3.0.0",
     "anndata>=0.12.0",
     "cp-measure>=0.1.13",
     "pycytominer>=1.4",
     "toml>=0.10",
     "matplotlib>=3.10",
+    "torch>=2.7.0,<3",
+    "torchvision>=0.22.0,<1",
 ]
 
 [project.scripts]
 feature_blocks = "feature_blocks.main:app"
 
 [project.optional-dependencies]
-cpu = [
-  "torch>=2.7.0,<3",
-  "torchvision>=0.22.0,<1",
-]
-cu128 = [
-  "torch>=2.7.0,<3",
-  "torchvision>=0.22.0,<1",
-]
 benchmarking = [
   "jupyterlab>=4.5.0",
   "scanpy[leiden]",
   "spatialdata_plot",
 ]
 
 [tool.uv]
-environments = [
-  "sys_platform == 'linux'",
-  "sys_platform == 'darwin'",
-]
-conflicts = [
-  [
-    { extra = "cpu" },
-    { extra = "cu128" },
-  ],
-]
 cache-keys = [
-  { file = "pyproject.toml" }, 
+  { file = "pyproject.toml" },
   { file = "**/**.py.txt" }, # Ensure rebuilds are triggered upon code changes
   { git = { commit = true } }
-]
-
-[tool.uv.sources]
-torch = [
-  { index = "pytorch-cpu", extra = "cpu" },
-  { index = "pytorch-cu128", extra = "cu128" },
-]
-torchvision = [
-  { index = "pytorch-cpu", extra = "cpu" },
-  { index = "pytorch-cu128", extra = "cu128" },
-]
-
-[[tool.uv.index]]
-name = "pytorch-cpu"
-url = "https://download.pytorch.org/whl/cpu"
-explicit = true
-
-[[tool.uv.index]]
-name = "pytorch-cu128"
-url = "https://download.pytorch.org/whl/cu128"
-explicit = true
+]