gin config reset for the tests.

google · Dec 4, 2024 · c293a77 · c293a77
1 parent def5bf8
commit c293a77
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/compiler_opt/rl/generate_bc_trajectories.py b/compiler_opt/rl/generate_bc_trajectories.py
@@ -339,6 +339,7 @@ def __init__(
     self._reward_key = reward_key
     kwargs.pop('reward_key', None)
     gin_config_str = kwargs.pop('gin_config_str', None)
+    gin.clear_config()
     gin.parse_config(gin_config_str)
     self._working_dir = None