Add flax installation in daily doctest workflow (#25860)

fix Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>

Add flax installation in daily doctest workflow (#25860)
fix Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
62399d6f · Yih-Dar · GitHub · 52574026 · 62399d6f
Unverified Commit 62399d6f authored Aug 30, 2023 by Yih-Dar Committed by GitHub Aug 30, 2023
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

.github/workflows/doctests.yml .github/workflows/doctests.yml +1 -1

No files found.
--- a/.github/workflows/doctests.yml
+++ b/.github/workflows/doctests.yml
@@ -34,7 +34,7 @@ jobs:
          nvidia-smi

      - name: Install transformers in edit mode
-        run: python3 -m pip install -e .
+        run: python3 -m pip install -e .[flax]

      - name: GPU visibility
        run: |