"tests/kernels/attention/test_attention.py" did not exist on "6e01e8c1c8ea323d30e3f57050469b2df66b56c6"
mypy.sh 557 Bytes
Newer Older
1
2
3
4
#!/bin/bash

CI=${1:-0}

5
6
7
8
if [ $CI -eq 1 ]; then
    set -e
fi

9
10
11
12
13
14
15
16
17
18
19
20
run_mypy() {
    echo "Running mypy on $1"
    if [ $CI -eq 1 ] && [ -z "$1" ]; then
        mypy "$@"
        return
    fi
    mypy --follow-imports skip "$@"
}

run_mypy # Note that this is less strict than CI
run_mypy tests
run_mypy vllm/attention
21
run_mypy vllm/compilation
22
23
24
25
26
27
28
29
30
run_mypy vllm/distributed
run_mypy vllm/engine
run_mypy vllm/executor
run_mypy vllm/lora
run_mypy vllm/model_executor
run_mypy vllm/plugins
run_mypy vllm/prompt_adapter
run_mypy vllm/spec_decode
run_mypy vllm/worker