FilesExpand file tree

master

/

pr-sanity-check.sh

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

60 lines (50 loc) · 1.72 KB

master

/

pr-sanity-check.sh

Copy path

Top

File metadata and controls

• Code
• Blame

60 lines (50 loc) · 1.72 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

#!/usr/bin/env bash

set -eou pipefail

GIT_TOP_DIR=$(git rev-parse --show-toplevel)

TMPFILE=$(mktemp)

trap "rm -rf ${TMPFILE}" EXIT

# By default just run against the latest commit

BASE=${BASE:-HEAD~1}

HEAD=${HEAD:-HEAD}

ancestor=$(git merge-base "${BASE}" "${HEAD}")

echo "INFO: Checking aginst the following stats"

(

set -x

git diff --stat "$ancestor" "${HEAD}" | sed '$d' > "${TMPFILE}"

)

while read -r git_attribute; do

if echo "${git_attribute}" | grep "linguist-generated=true" >/dev/null 2>/dev/null; then

pattern=$(echo ${git_attribute} | cut -d' ' -f1)

escaped_pattern=$(printf '%s\n' "$pattern" | sed -e 's/[\/&]/\\&/g')

# Delete known generated files

sed -i '/'"${escaped_pattern}"'/d' "${TMPFILE}"

fi

done < "${GIT_TOP_DIR}/.gitattributes"

echo "INFO: Showing non-generated files:"

(

set -x

cat "${TMPFILE}"

)

# Get only files that have changed

changed_files=$(cut -d' ' -f2 "${TMPFILE}" | xargs)

details=$(git diff --shortstat "$ancestor" "${HEAD}" -- ${changed_files})

add=$(echo "$details" | grep -o '[0-9]* insertion' | grep -o '[0-9]*' || true)

remove=$(echo "$details" | grep -o '[0-9]* deletion' | grep -o '[0-9]*' || true)

pr_size=0

if [ "$add" ]; then

pr_size=$(("$pr_size" + "$add"))

fi

if [ "$remove" ]; then

pr_size=$(("$pr_size" + "$remove"))

fi

echo "INFO: PR SIZE is ${pr_size}"

if ((pr_size > 2000)); then

echo

echo 'Your PR is '"$pr_size"' LOC which is more than the 2000 maximum'

echo 'allowed within PyTorch infra. PLease make sure to split up'

echo 'your PR into smaller pieces that can be reviewed.'

echo 'If you think that this rule should not apply to your PR,'

echo 'please contact @albanD or @seemethere.'

echo

exit 1

fi