Use torch+cpu to cut venv size

Author: elvout

README.md

@@ -10,7 +10,7 @@
 
 * Python 3.10
 * uv package manager
-* ~10 GB disk space for Python packages
+* ~2 GB disk space for Python packages
 
 ### Extra Dependencies for Deployment and Data Analysis
 

pyproject.toml

@@ -14,7 +14,7 @@ dependencies = [
 
 [dependency-groups]
 dev = ["mypy", "jupyter", "pandas-stubs", "types-tqdm", "types-seaborn"]
-deployment = ["pytorch-mppi==0.7.5", "cooperative_transport"]
+deployment = ["torch", "pytorch-mppi==0.7.5", "cooperative_transport"]
 evaluation = [
     "rosbags",
     "matplotlib",
@@ -32,7 +32,12 @@ evaluation = [
 
 [tool.uv.sources]
 cooperative_transport = { path = "src/eleyng/table-carrying-ai", editable = true }
+torch = { index = "pytorch-cpu" }
 
+[[tool.uv.index]]
+name = "pytorch-cpu"
+url = "https://download.pytorch.org/whl/cpu"
+explicit = true
 
 [build-system]
 requires = ["hatchling"]