wehub-resource-sync
92 lines
2.5 KiB
Python
92 lines
2.5 KiB
Python
# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
import argparse
|
|
import json
|
|
import os
|
|
import re
|
|
import sys
|
|
|
|
import requests
|
|
|
|
SKIP_COVERAGE_CHECKING_LABELS = [
|
|
"cinn",
|
|
"typing",
|
|
"codestyle",
|
|
"phi_ops",
|
|
r"3.1\d", # SOT Python 3.13 support
|
|
]
|
|
|
|
SKIP_DISTRIBUTE_TEST_CHECKING_LABELS = [
|
|
"cinn",
|
|
"BUAA",
|
|
]
|
|
|
|
SKIP_COVERAGE_CHECKING_REGEX = re.compile(
|
|
rf"[\[【]({'|'.join(SKIP_COVERAGE_CHECKING_LABELS)})[\]】]",
|
|
re.IGNORECASE,
|
|
)
|
|
|
|
SKIP_DISTRIBUTE_TEST_CHECKING_REGEX = re.compile(
|
|
rf"[\[【]({'|'.join(SKIP_DISTRIBUTE_TEST_CHECKING_LABELS)})[\]】]",
|
|
re.IGNORECASE,
|
|
)
|
|
|
|
|
|
def check_skip_coverage(title):
|
|
if match_obj := SKIP_COVERAGE_CHECKING_REGEX.search(title):
|
|
print(f'The title starts with {match_obj.group(0)}')
|
|
else:
|
|
print(
|
|
f'The title does not start with {" or ".join(SKIP_COVERAGE_CHECKING_LABELS)}'
|
|
)
|
|
sys.exit(1)
|
|
|
|
|
|
def check_skip_distribute_test(title):
|
|
if match_obj := SKIP_DISTRIBUTE_TEST_CHECKING_REGEX.search(title):
|
|
print(f'The title starts with {match_obj.group(0)}')
|
|
else:
|
|
print(
|
|
f'The title does not start with {" or ".join(SKIP_COVERAGE_CHECKING_LABELS)}'
|
|
)
|
|
sys.exit(1)
|
|
|
|
|
|
def main(args):
|
|
pr_id = os.getenv('GIT_PR_ID')
|
|
if not pr_id:
|
|
print('PREC No PR ID')
|
|
sys.exit(1)
|
|
response = requests.get(
|
|
f'https://api.github.com/repos/PaddlePaddle/Paddle/pulls/{pr_id}',
|
|
headers={'Accept': 'application/vnd.github.v3+json'},
|
|
)
|
|
|
|
data = json.loads(response.text)
|
|
|
|
title = data['title']
|
|
if args.type == 'skip_coverage_check':
|
|
check_skip_coverage(title)
|
|
elif args.type == 'skip_distribute_test':
|
|
print(check_skip_distribute_test(title))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description='Process some arguments.')
|
|
parser.add_argument('type', help='The type of check to perform.')
|
|
|
|
args = parser.parse_args()
|
|
main(args)
|