Revert previous two commits

Revert "Revert "Accept decay rate parameter in MockCompilationRunner""

This reverts commit 94cb0d6.

HEAD~1 != HEAD...

Revert "Remove flag allow_override calls"

This reverts commit 7091b15.

See previous commit message.

Author: boomanaiden154

compiler_opt/rl/imitation_learning/generate_bc_trajectories_test.py

@@ -33,6 +33,9 @@
 from compiler_opt.rl import env
 from compiler_opt.rl import env_test
 
+# flags.FLAGS['gin_files'].allow_override = True
+# flags.FLAGS['gin_bindings'].allow_override = True
+
 _eps = 1e-5
 
 

compiler_opt/tools/generate_default_trace_test.py

@@ -16,6 +16,7 @@
 import os
 from unittest import mock
 
+from absl import flags
 from absl.testing import absltest
 from absl.testing import flagsaver
 import gin
@@ -28,12 +29,17 @@
 
 from tf_agents.system import system_multiprocessing as multiprocessing
 
+flags.FLAGS['num_workers'].allow_override = True
+flags.FLAGS['gin_files'].allow_override = True
+flags.FLAGS['gin_bindings'].allow_override = True
+
 
 @gin.configurable(module='runners')
 class MockCompilationRunner(compilation_runner.CompilationRunner):
   """A compilation runner just for test."""
 
-  def __init__(self, sentinel=None):
+  def __init__(self, moving_average_decay_rate: float, sentinel=None):
+    del moving_average_decay_rate  # Unused.
     assert sentinel == 42
     super().__init__()
 

compiler_opt/tools/generate_test_model_test.py

@@ -16,13 +16,17 @@
 An integration test for model saving, to detect TFLite model conversion.
 """
 
+from absl import flags
 from absl.testing import absltest
 from absl.testing import flagsaver
 from absl.testing import parameterized
 import gin
 
 from compiler_opt.tools import generate_test_model
 
+flags.FLAGS['gin_files'].allow_override = True
+flags.FLAGS['gin_bindings'].allow_override = True
+
 
 def _get_test_settings():
   test_setting = []