FilesExpand file tree

main

/

pr_flax_dependency_test.yml

Copy path

View runs

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

38 lines (35 loc) · 1.06 KB

main

/

pr_flax_dependency_test.yml

Copy path

Top

File metadata and controls

• Code
• Blame

38 lines (35 loc) · 1.06 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

name: Run Flax dependency tests

on:

pull_request:

branches:

- main

paths:

- "src/diffusers/**.py"

push:

branches:

- main

concurrency:

group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}

cancel-in-progress: true

jobs:

check_flax_dependencies:

runs-on: ubuntu-latest

steps:

- uses: actions/checkout@v3

- name: Set up Python

uses: actions/setup-python@v4

with:

python-version: "3.8"

- name: Install dependencies

run: |

python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"

python -m pip install --upgrade pip uv

python -m uv pip install -e .

python -m uv pip install "jax[cpu]>=0.2.16,!=0.3.2"

python -m uv pip install "flax>=0.4.1"

python -m uv pip install "jaxlib>=0.1.65"

python -m uv pip install pytest

- name: Check for soft dependencies

run: |

python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"

pytest tests/others/test_dependencies.py