From e7532c9b0c85edc6dc92966633fa86a52b099cd4 Mon Sep 17 00:00:00 2001
From: Wing Lian <wing@axolotl.ai>
Date: Fri, 21 Mar 2025 06:57:06 -0400
Subject: [PATCH] make sure ninja is installed

---
 cicd/Dockerfile.jinja | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/cicd/Dockerfile.jinja b/cicd/Dockerfile.jinja
index 35eccb495..da3e5a8ad 100644
--- a/cicd/Dockerfile.jinja
+++ b/cicd/Dockerfile.jinja
@@ -31,9 +31,9 @@ RUN if [ "$NIGHTLY_BUILD" = "true" ] ; then \
         sed -i 's#^datasets.*#datasets @ git+https://github.com/huggingface/datasets.git@main#' requirements.txt; \
     fi
 
-RUN pip3 install -U packaging setuptools wheel
+RUN pip3 install -U packaging setuptools wheel ninja
 RUN if [ "$AXOLOTL_EXTRAS" != "" ] ; then \
-        pip install deepspeed; \
+        pip install --no-build-isolation deepspeed; \
         pip install --no-build-isolation -e .[flash-attn,optimizers,ray,$AXOLOTL_EXTRAS] $AXOLOTL_ARGS; \
     else \
         pip install deepspeed; \