From d86cb8d0fdb0c687574eda27beba800e294a74e1 Mon Sep 17 00:00:00 2001
From: Victor Giers <mail@victorgiers.com>
Date: Fri, 20 Mar 2026 10:09:09 +0100
Subject: [PATCH] Enhance run.sh to handle legacy Python state checks

---
 run.sh | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/run.sh b/run.sh
index dbb7ac4..07aefe5 100755
--- a/run.sh
+++ b/run.sh
@@ -208,7 +208,15 @@ fi
 printf '%s\n' "$TORCH_FLAVOR" > "$TORCH_FLAVOR_FILE"
 echo "Using PyTorch flavor: $TORCH_FLAVOR"
 
-if [ "$RECREATE_VENV" -eq 0 ] && python_deps_usable; then
+NEEDS_LEGACY_PYTHON_STATE=0
+if [ ! -r "$PYTHON_DEPS_STATE_FILE" ]; then
+  NEEDS_LEGACY_PYTHON_STATE=1
+fi
+if [ -z "$HEIMGEIST_TORCH_INDEX_URL" ] && [ ! -r "$TORCH_STATE_FILE" ]; then
+  NEEDS_LEGACY_PYTHON_STATE=1
+fi
+
+if [ "$RECREATE_VENV" -eq 0 ] && [ "$NEEDS_LEGACY_PYTHON_STATE" -eq 1 ] && python_deps_usable; then
   if [ ! -r "$PYTHON_DEPS_STATE_FILE" ]; then
     write_state "$PYTHON_DEPS_STATE_FILE" backend/requirements.txt
   fi