Merge branch 'policy_combiner' of https://github.com/tvmarino/ml-comp…

…iler-opt into policy_combiner
google · Sep 6, 2024 · bf12cee · bf12cee
2 parents 47f5efc + 86e4d12
commit bf12cee
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/compiler_opt/tools/combine_tfa_policies_lib_test.py b/compiler_opt/tools/combine_tfa_policies_lib_test.py
@@ -40,7 +40,7 @@ def __init__(self):
     super(AddOnePolicy, self).__init__(
         time_step_spec=time_step_spec, action_spec=act_spec)
 
-  def _distribution(self, t_step):
+  def _distribution(self, time_step):
     pass
 
   def _variables(self):
@@ -66,7 +66,7 @@ def __init__(self):
     super(SubtractOnePolicy, self).__init__(
         time_step_spec=time_step_spec, action_spec=act_spec)
 
-  def _distribution(self, t_step):
+  def _distribution(self, time_step):
     pass
 
   def _variables(self):