FilesExpand file tree

main

/

run_tests_from_a_pr.yml

Copy path

View runs

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

74 lines (64 loc) · 2 KB

main

/

run_tests_from_a_pr.yml

Copy path

Top

File metadata and controls

• Code
• Blame

74 lines (64 loc) · 2 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

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

name: Check running SLOW tests from a PR (only GPU)

on:

workflow_dispatch:

inputs:

docker_image:

default: 'diffusers/diffusers-pytorch-cuda'

description: 'Name of the Docker image'

required: true

pr_number:

description: 'PR number to test on'

required: true

test:

description: 'Tests to run (e.g.: `tests/models`).'

required: true

env:

DIFFUSERS_IS_CI: yes

IS_GITHUB_CI: "1"

HF_HOME: /mnt/cache

OMP_NUM_THREADS: 8

MKL_NUM_THREADS: 8

PYTEST_TIMEOUT: 600

RUN_SLOW: yes

jobs:

run_tests:

name: "Run a test on our runner from a PR"

runs-on:

group: aws-g4dn-2xlarge

container:

image: ${{ github.event.inputs.docker_image }}

options: --gpus 0 --privileged --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/

steps:

- name: Validate test files input

id: validate_test_files

env:

PY_TEST: ${{ github.event.inputs.test }}

run: |

if [[ ! "$PY_TEST" =~ ^tests/ ]]; then

echo "Error: The input string must start with 'tests/'."

exit 1

fi

if [[ ! "$PY_TEST" =~ ^tests/(models|pipelines|lora) ]]; then

echo "Error: The input string must contain either 'models', 'pipelines', or 'lora' after 'tests/'."

exit 1

fi

if [[ "$PY_TEST" == *";"* ]]; then

echo "Error: The input string must not contain ';'."

exit 1

fi

echo "$PY_TEST"

shell: bash -e {0}

- name: Checkout PR branch

uses: actions/checkout@v4

with:

ref: refs/pull/${{ inputs.pr_number }}/head

- name: Install pytest

run: |

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

python -m uv pip install -e [quality,test]

python -m uv pip install peft

- name: Run tests

env:

PY_TEST: ${{ github.event.inputs.test }}

run: |

pytest "$PY_TEST"