Another fix

Author: mseeger

.github/workflows/unit-tests.yml

@@ -32,7 +32,7 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        python -m pip install 'litgpt[all,test]'
+        python -m pip install 'litgpt[all,test,extra]'
         python -m pip install .
     - name: Test with pytest
       run: |