Skip to content

Commit 37a6c90

Browse files
Accept decay rate parameter in MockCompilationRunner
The mock used in generate_default_trace_test needs to accept moving_average_decay_rate as it is automatically passed in by the instantiation setup in generate_trace in generate_default_trace.py. The test still passes if it isn't this way, but doesn't actually "compile" any modules due to the errors, which sort of defeats the point.
1 parent 4c590d6 commit 37a6c90

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

compiler_opt/tools/generate_default_trace_test.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,8 @@
3838
class MockCompilationRunner(compilation_runner.CompilationRunner):
3939
"""A compilation runner just for test."""
4040

41-
def __init__(self, sentinel=None):
41+
def __init__(self, moving_average_decay_rate: float, sentinel=None):
42+
del moving_average_decay_rate # Unused.
4243
assert sentinel == 42
4344
super().__init__()
4445

0 commit comments

Comments
 (0)