From 980e7aa44d667b9cbbfe01b9743edb00d0ac447b Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Wed, 7 Feb 2024 09:43:31 -0500
Subject: [PATCH] install packaging for various tests

---
 .github/workflows/tests.yml | 1 +
 docker/Dockerfile           | 3 ++-
 docker/Dockerfile-tests     | 3 +++
 3 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index b14d080ba..68ca8534c 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -48,6 +48,7 @@ jobs:
 
       - name: Install dependencies
         run: |
+          pip3 install --upgrade pip
           pip3 install packaging
           pip3 install -U -e .
           pip3 install -r requirements-tests.txt
diff --git a/docker/Dockerfile b/docker/Dockerfile
index f8b97771d..f59033c7b 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -18,7 +18,8 @@ RUN git clone --depth=1 https://github.com/OpenAccess-AI-Collective/axolotl.git
 
 WORKDIR /workspace/axolotl
 
-RUN pip install packaging
+RUN pip install --upgrade pip && \
+    pip install packaging
 
 # If AXOLOTL_EXTRAS is set, append it in brackets
 RUN if [ "$AXOLOTL_EXTRAS" != "" ] ; then \
diff --git a/docker/Dockerfile-tests b/docker/Dockerfile-tests
index 2ec94f868..585f68ba3 100644
--- a/docker/Dockerfile-tests
+++ b/docker/Dockerfile-tests
@@ -22,6 +22,9 @@ WORKDIR /workspace/axolotl
 RUN git fetch origin +$GITHUB_REF && \
     git checkout FETCH_HEAD
 
+RUN pip install --upgrade pip && \
+    pip install packaging \
+
 # If AXOLOTL_EXTRAS is set, append it in brackets
 RUN if [ "$AXOLOTL_EXTRAS" != "" ] ; then \
         pip install -e .[deepspeed,flash-attn,mamba-ssm,$AXOLOTL_EXTRAS]; \