From 4609e3b166ff0ce8e926f39d541aa7ef76592ec4 Mon Sep 17 00:00:00 2001
From: Maxime <672982+maximegmd@users.noreply.github.com>
Date: Tue, 6 Feb 2024 16:38:00 +0100
Subject: [PATCH] CI fixes

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

diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index 2d9969524..b14d080ba 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/docker/Dockerfile b/docker/Dockerfile
index efc40ab06..f8b97771d 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -18,6 +18,8 @@ RUN git clone --depth=1 https://github.com/OpenAccess-AI-Collective/axolotl.git
 
 WORKDIR /workspace/axolotl
 
+RUN 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]; \