From 518f2864154cf40b40e9a3252665ed105b57585b Mon Sep 17 00:00:00 2001
From: jeffquinn-msk <105746884+jeffquinn-msk@users.noreply.github.com>
Date: Thu, 24 Oct 2024 13:48:18 -0400
Subject: [PATCH] Add Memray, Change to PyTorch base image (#201)

---
 Dockerfile     | 11 +++++++----
 pyproject.toml |  2 +-
 2 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index ad86c64..488dd38 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,13 +1,16 @@
-FROM python:3.11
+FROM pytorch/pytorch:latest
+
+USER root
+
+RUN --mount=type=cache,target=/.cache/pip pip install --upgrade pip
+RUN --mount=type=cache,target=/.cache/pip pip install ipython memray
 
-RUN --mount=type=cache,target=/root/.cache pip install --upgrade pip
-RUN --mount=type=cache,target=/root/.cache pip install ipython
 
 RUN mkdir /app
 COPY src/ /app/src/
 COPY tox.ini pyproject.toml setup.py setup.cfg LICENSE.txt README.md /app/
 
-RUN --mount=type=cache,target=/root/.cache cd /app && pip install '.[dev,test]' --extra-index-url https://download.pytorch.org/whl/cpu
+RUN --mount=type=cache,target=/.cache/pip cd /app && pip install '.[dev,test]'
 
 ENV PYTHONUNBUFFERED=1
 
diff --git a/pyproject.toml b/pyproject.toml
index 3f920c2..7328935 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -3,7 +3,7 @@ name = "bayestme"
 version = "1.0.0"
 description = "A reference-free Bayesian method that discovers spatial transcriptional programs in the tissue microenvironment"
 readme = "README.md"
-requires-python = ">=3.11"
+requires-python = ">=3.10"
 license = {file = "LICENSE.txt"}
 keywords = ["spatial", "expression", "rna"]
 authors = [