From b5b7f565d333bfb11f51fee6bd1cbf2dd637d9cc Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Fri, 29 Mar 2024 00:50:43 -0400
Subject: [PATCH] install packaging for conv1d package

---
 .github/workflows/pypi.yml  | 2 +-
 .github/workflows/tests.yml | 1 +
 cicd/Dockerfile.jinja       | 1 +
 docker/Dockerfile           | 1 +
 4 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/pypi.yml b/.github/workflows/pypi.yml
index dbd225f6f0..885239d185 100644
--- a/.github/workflows/pypi.yml
+++ b/.github/workflows/pypi.yml
@@ -25,7 +25,7 @@ jobs:
 
       - name: Install dependencies
         run: |
-          pip3 install wheel
+          pip3 install wheel packaging
           pip3 install -e .
           pip3 install -r requirements-tests.txt
 
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index d8ca5400a9..769287b710 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -48,6 +48,7 @@ jobs:
 
       - name: Install dependencies
         run: |
+          pip3 install packaging
           pip3 install -U -e .
           pip3 install -r requirements-tests.txt
 
diff --git a/cicd/Dockerfile.jinja b/cicd/Dockerfile.jinja
index 19d7a09de5..56af119c01 100644
--- a/cicd/Dockerfile.jinja
+++ b/cicd/Dockerfile.jinja
@@ -22,6 +22,7 @@ RUN git fetch origin +$GITHUB_REF && \
     git checkout FETCH_HEAD
 
 # If AXOLOTL_EXTRAS is set, append it in brackets
+RUN pip install packaging
 RUN if [ "$AXOLOTL_EXTRAS" != "" ] ; then \
         pip install -e .[deepspeed,flash-attn,mamba-ssm,galore,$AXOLOTL_EXTRAS] $AXOLOTL_ARGS; \
     else \
diff --git a/docker/Dockerfile b/docker/Dockerfile
index 84046c6522..eb2e33d3d6 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -20,6 +20,7 @@ RUN git clone --depth=1 https://github.com/OpenAccess-AI-Collective/axolotl.git
 WORKDIR /workspace/axolotl
 
 # If AXOLOTL_EXTRAS is set, append it in brackets
+RUN pip install packaging
 RUN if [ "$AXOLOTL_EXTRAS" != "" ] ; then \
         pip install -e .[deepspeed,flash-attn,mamba-ssm,galore,$AXOLOTL_EXTRAS] $AXOLOTL_ARGS; \
     else \