diff --git a/tools/get_pr_ut.py b/tools/get_pr_ut.py index 0ba60265353073a1a062583632f3f25290f3a1e3..6b90a656f0107c5ee40812d329eab94173885ecd 100644 --- a/tools/get_pr_ut.py +++ b/tools/get_pr_ut.py @@ -139,13 +139,10 @@ class PRChecker(object): """ judge is white file in pr's files. """ isWhiteFile = False not_white_files = (PADDLE_ROOT + 'cmake/', PADDLE_ROOT + 'patches/', - PADDLE_ROOT + 'paddle/testing/', PADDLE_ROOT + 'tools/dockerfile/', PADDLE_ROOT + 'tools/windows/', PADDLE_ROOT + 'tools/test_runner.py', - PADDLE_ROOT + 'tools/parallel_UT_rule.py', - PADDLE_ROOT + 'paddle/scripts/paddle_build.sh', - PADDLE_ROOT + 'paddle/scripts/paddle_build.bat') + PADDLE_ROOT + 'tools/parallel_UT_rule.py') if 'cmakelist' in filename.lower(): isWhiteFile = False elif filename.startswith((not_white_files)): @@ -285,9 +282,21 @@ class PRChecker(object): file_list = [] file_dict = self.get_pr_files() for filename in file_dict: - if filename.startswith( - (PADDLE_ROOT + 'python/', PADDLE_ROOT + 'paddle/fluid/')): + if filename.startswith(PADDLE_ROOT + 'python/'): file_list.append(filename) + elif filename.startswith(PADDLE_ROOT + 'paddle/'): + if filename.startswith((PADDLE_ROOT + 'paddle/infrt', + PADDLE_ROOT + 'paddle/utils')): + filterFiles.append(filename) + elif filename.startswith(PADDLE_ROOT + 'paddle/scripts'): + if filename.startswith( + (PADDLE_ROOT + 'paddle/scripts/paddle_build.sh', + PADDLE_ROOT + 'paddle/scripts/paddle_build.bat')): + file_list.append(filename) + else: + filterFiles.append(filename) + else: + file_list.append(filename) else: if file_dict[filename] == 'added': file_list.append(filename)