yapf . -ir

google · Sep 6, 2024 · 5d6783d · 5d6783d
1 parent 35d9e8c
commit 5d6783d
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/compiler_opt/tools/combine_tfa_policies_lib_test.py b/compiler_opt/tools/combine_tfa_policies_lib_test.py
@@ -35,8 +35,7 @@ def __init__(self):
 
     act_spec = tensor_spec.TensorSpec(shape=(1,), dtype=tf.int64)
 
-    super().__init__(
-        time_step_spec=time_step_spec, action_spec=act_spec)
+    super().__init__(time_step_spec=time_step_spec, action_spec=act_spec)
 
   def _distribution(self, time_step):
     pass
@@ -59,8 +58,7 @@ def __init__(self):
 
     act_spec = tensor_spec.TensorSpec(shape=(1,), dtype=tf.int64)
 
-    super().__init__(
-        time_step_spec=time_step_spec, action_spec=act_spec)
+    super().__init__(time_step_spec=time_step_spec, action_spec=act_spec)
 
   def _distribution(self, time_step):
     pass