Refine precision test message

4 years ago · 49420eb047
parent 7c8ac064c8
commit 49420eb047
1 changed files with 5 additions and 5 deletions
--- a/tools/get_pr_ut.py
+++ b/tools/get_pr_ut.py
@ -77,12 +77,12 @@ class PRChecker(object):
                ut_list.append(ut.rstrip('\r\n'))

        if check_added_ut:
-            cmd = 'bash {}/tools/check_added_ut.sh'.format(PADDLE_ROOT)
+            cmd = 'bash {}/tools/check_added_ut.sh >/tmp/pre_ut 2>&1'.format(
+                PADDLE_ROOT)
            os.system(cmd)
-
-        with open('{}/added_ut'.format(PADDLE_ROOT)) as utfile:
-            for ut in utfile:
-                ut_list.append(ut.rstrip('\r\n'))
+            with open('{}/added_ut'.format(PADDLE_ROOT)) as utfile:
+                for ut in utfile:
+                    ut_list.append(ut.rstrip('\r\n'))

        return ' '.join(ut_list)