From ce4111905135d1dde0b76031f3dc39ee60daf33c Mon Sep 17 00:00:00 2001 From: Dave Halter Date: Thu, 16 Mar 2017 21:45:51 +0100 Subject: [PATCH] Fix some stuff in a diff profile test script. --- scripts/diff_parser_profile.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/scripts/diff_parser_profile.py b/scripts/diff_parser_profile.py index ec37e53b..3a29afee 100755 --- a/scripts/diff_parser_profile.py +++ b/scripts/diff_parser_profile.py @@ -9,15 +9,15 @@ Usage: Options: -h --help Show this screen. -d --debug Enable Jedi internal debugging. - -s Sort the profile results, e.g. cum, name [default: time]. + -s Sort the profile results, e.g. cumtime, name [default: time]. """ import cProfile from docopt import docopt -from jedi.parser import load_grammar +from jedi.parser.python import load_grammar from jedi.parser.diff import DiffParser -from jedi.parser import ParserWithRecovery +from jedi.parser.python import ParserWithRecovery from jedi._compatibility import u from jedi.common import splitlines import jedi @@ -29,7 +29,9 @@ def run(parser, lines): def main(args): - jedi.set_debug_function(notices=args['--debug']) + if args['--debug']: + jedi.set_debug_function(notices=True) + with open(args['']) as f: code = f.read() grammar = load_grammar()