Fixed Aiden's nits.

google · Sep 9, 2024 · 78460ce · 78460ce
1 parent 3b0cefd
commit 78460ce
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 9 deletions.
diff --git a/compiler_opt/tools/combine_tfa_policies.py b/compiler_opt/tools/combine_tfa_policies.py
@@ -28,15 +28,11 @@
 from compiler_opt.tools import combine_tfa_policies_lib as cfa_lib
 
 _COMBINE_POLICIES_NAMES = flags.DEFINE_multi_string(
-    'policies_names',
-    [],
-    'List in order of policy names for combined policies. Order must match that of policies_paths.'  # pylint: disable=line-too-long
-)
+    'policies_names', [], 'List in order of policy names for combined policies.'
+    'Order must match that of policies_paths.')
 _COMBINE_POLICIES_PATHS = flags.DEFINE_multi_string(
-    'policies_paths',
-    [],
-    'List in order of policy paths for combined policies. Order must match that of policies_names.'  # pylint: disable=line-too-long
-)
+    'policies_paths', [], 'List in order of policy paths for combined policies.'
+    'Order must match that of policies_names.')
 _COMBINED_POLICY_PATH = flags.DEFINE_string(
     'combined_policy_path', '', 'Path to save the combined policy.')
 _GIN_FILES = flags.DEFINE_multi_string(

diff --git a/compiler_opt/tools/combine_tfa_policies_lib.py b/compiler_opt/tools/combine_tfa_policies_lib.py
@@ -93,7 +93,7 @@ def _action(self,
         discount=time_step.discount,
         observation=new_observation)
 
-    # TODO(359): We only support combining two policies.Generalize this to
+    # TODO(359): We only support combining two policies. Generalize this to
     # handle multiple policies.
     def f0():
       return tf.cast(