diff options
Diffstat (limited to 'startop/scripts/iorap/compiler.py')
| -rwxr-xr-x | startop/scripts/iorap/compiler.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/startop/scripts/iorap/compiler.py b/startop/scripts/iorap/compiler.py index a335b30da8a5..ef9b870b8113 100755 --- a/startop/scripts/iorap/compiler.py +++ b/startop/scripts/iorap/compiler.py @@ -31,8 +31,10 @@ import tempfile from pathlib import Path from typing import Iterable, Optional, List -from generated.TraceFile_pb2 import * -from lib.inode2filename import Inode2Filename +DIR = os.path.abspath(os.path.dirname(__file__)) +sys.path.append(os.path.dirname(DIR)) +from iorap.generated.TraceFile_pb2 import * +from iorap.lib.inode2filename import Inode2Filename parent_dir_name = os.path.dirname(os.path.dirname(os.path.realpath(__file__))) sys.path.append(parent_dir_name) @@ -305,7 +307,7 @@ def main(argv): transform_perfetto_trace_to_systrace(options.perfetto_trace_file, trace_file.name) return run(sql_db_path, - options.trace_file, + trace_file.name, options.trace_duration, options.output_file, inode_table, |