diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index 6ca9443..1825711 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -22,7 +22,7 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          pip install .[testing,quality]
+          pip install .[training,testing,quality]
 
       - name: Check isort
         uses: isort/isort-action@v1