From 617e3782163b755c151c7a94d970cbb02bc63fce Mon Sep 17 00:00:00 2001
From: Vladislav <vkozlov@nvidia.com>
Date: Tue, 7 May 2024 03:30:34 -0600
Subject: [PATCH] Add missing python packages for levanter tests (#804)

---
 .github/workflows/_ci.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/_ci.yaml b/.github/workflows/_ci.yaml
index 3ee16ae9d..c16f82dbc 100644
--- a/.github/workflows/_ci.yaml
+++ b/.github/workflows/_ci.yaml
@@ -405,7 +405,7 @@ jobs:
         docker run -i --gpus all --shm-size=1g \
         ${{ needs.build-levanter.outputs.DOCKER_TAG_FINAL }} \
         bash <<"EOF" |& tee test-levanter.log
-          pip install pytest
+          pip install flake8 pytest soundfile librosa
           PYTHONPATH=/opt/levanter/tests:$PYTHONPATH pytest /opt/levanter/tests
         EOF
       STATISTICS_SCRIPT: |