From fa715a2ff34f77a7d41c6fcf500555e5396b9199 Mon Sep 17 00:00:00 2001
From: Nikhil Woodruff <nikhil.woodruff@outlook.com>
Date: Mon, 16 Sep 2024 17:51:58 +0100
Subject: [PATCH] Remove torch from required

---
 Makefile               | 3 +++
 docker/docs.Dockerfile | 2 +-
 pyproject.toml         | 4 ++--
 3 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/Makefile b/Makefile
index b254604..2fca225 100644
--- a/Makefile
+++ b/Makefile
@@ -9,6 +9,9 @@ test:
 install:
 	pip install -e ".[dev]"
 
+install-lite:
+	pip install -e .
+
 download:
 	python policyengine_uk_data/storage/download_private_prerequisites.py
 
diff --git a/docker/docs.Dockerfile b/docker/docs.Dockerfile
index 090ca7e..4463c58 100644
--- a/docker/docs.Dockerfile
+++ b/docker/docs.Dockerfile
@@ -1,5 +1,5 @@
 FROM python:latest
 COPY . .
-RUN make install
+RUN make install-lite
 EXPOSE 8080
 ENTRYPOINT ["streamlit", "run", "docs/Home.py", "--server.port=8080", "--server.address=0.0.0.0"]
diff --git a/pyproject.toml b/pyproject.toml
index 640810d..6142ddd 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -15,8 +15,6 @@ requires-python = ">=3.6"
 dependencies = [
     "policyengine_core",
     "tables",
-    "survey_enhance",
-    "torch",
     "requests",
     "tqdm",
     "tabulate",
@@ -28,6 +26,8 @@ dev = [
     "black",
     "pytest",
     "policyengine_uk>=1.8.0",
+    "survey_enhance",
+    "torch",
     "streamlit",
 ]