Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-23 19:49:24 +01:00
Code Issues Packages Projects Releases Wiki Activity

remove profiling log (#113)

Browse Source 
UT test by default does not run profiling
This commit is contained in:
wenmeng zhou
2023-02-17 12:01:54 +08:00
committed by GitHub
parent 5df2930017
commit 3ca68bf22f
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/run.py
View File

@@ -600,7 +600,7 @@ if __name__ == '__main__':
parser.add_argument( parser.add_argument(
'--level', default=0, type=int, help='2 -- all, 1 -- p1, 0 -- p0') '--level', default=0, type=int, help='2 -- all, 1 -- p1, 0 -- p0')
parser.add_argument( parser.add_argument(
'--disable_profile', action='store_true', help='disable profiling') '--profile', action='store_true', help='enable profiling')
parser.add_argument( parser.add_argument(
'--run_config', '--run_config',
default=None, default=None,
@@ -634,7 +634,7 @@ if __name__ == '__main__':
set_test_level(args.level) set_test_level(args.level)
os.environ['REGRESSION_BASELINE'] = '1' os.environ['REGRESSION_BASELINE'] = '1'
logger.info(f'TEST LEVEL: {test_level()}') logger.info(f'TEST LEVEL: {test_level()}')
if not args.disable_profile: if args.profile:
from utils import profiler from utils import profiler
logger.info('enable profile ...') logger.info('enable profile ...')
profiler.enable() profiler.enable()