From 78460ce93ca26b495cc243949c3de1c980e346af Mon Sep 17 00:00:00 2001 From: "Teodor V. Marinov" Date: Mon, 9 Sep 2024 18:58:17 +0000 Subject: [PATCH] Fixed Aiden's nits. --- compiler_opt/tools/combine_tfa_policies.py | 12 ++++-------- compiler_opt/tools/combine_tfa_policies_lib.py | 2 +- 2 files changed, 5 insertions(+), 9 deletions(-) diff --git a/compiler_opt/tools/combine_tfa_policies.py b/compiler_opt/tools/combine_tfa_policies.py index 0aad21b9..c3146711 100755 --- 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 index 4a53ba19..4d1de2b3 100644 --- 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(