From cf65e488d877bd6d7bfa9723efc8a15ffa1c62e6 Mon Sep 17 00:00:00 2001
From: Roland Tannous <rolandtannous@gravityq.ai>
Date: Thu, 26 Mar 2026 03:55:35 +0000
Subject: [PATCH] install local branch in Docker setup.sh path

---
 studio/setup.sh | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/studio/setup.sh b/studio/setup.sh
index b254b84704..de1bccad5d 100755
--- a/studio/setup.sh
+++ b/studio/setup.sh
@@ -365,6 +365,11 @@ fi
 if [ "$_DOCKER_NO_VENV" = true ]; then
     echo "   Docker detected — skipping venv activation."
 
+    # Install branch's unsloth/unsloth_cli/studio into /opt/conda
+    # (overwrites PyPI version with Docker-aware code)
+    echo "   Installing local unsloth from branch..."
+    pip install --force-reinstall --no-deps "$REPO_ROOT"
+
     # Install only missing deps (studio, data-designer, plugin, metadata patch).
     # Heavy packages (torch, unsloth, vllm, etc.) are already in /opt/conda.
     # install_python_stack.py has steps 1-5 commented out for this branch.