Skip to content

Commit 2351816

Browse files
authored
[ir2vec] Make the ir2vec_vocab_path optional (#537)
1 parent 07d7ab4 commit 2351816

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

compiler_opt/rl/inlining/inlining_runner.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -39,9 +39,12 @@ class InliningRunner(compilation_runner.CompilationRunner):
3939
ir_path, tf_policy_path, default_reward, moving_average_reward)
4040
"""
4141

42-
def __init__(self, llvm_size_path: str, ir2vec_vocab_path: str | None, *args,
42+
def __init__(self,
43+
*,
44+
llvm_size_path: str,
45+
ir2vec_vocab_path: str | None = None,
4346
**kwargs):
44-
super().__init__(*args, **kwargs)
47+
super().__init__(**kwargs)
4548
self._llvm_size_path = llvm_size_path
4649
self._ir2vec_vocab_path = ir2vec_vocab_path
4750

0 commit comments

Comments
 (0)