diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index 46316e1018..9b1c5364a6 100644
--- a/.github/workflows/ci.yaml
+++ b/.github/workflows/ci.yaml
@@ -112,7 +112,7 @@ jobs:
         shell: bash # for Windows compatibility
         run: |
           python -m pip install --upgrade pip wheel
-          pip install -e ".[train, export]" "coverage[toml]" --extra-index-url https://download.pytorch.org/whl/cpu
+          pip install -e ".[export]" "coverage[toml]" --extra-index-url https://download.pytorch.org/whl/cpu
       - name: Check environment
         run: |
           yolo checks