Skip to content

Commit 0dbc785

Browse files
authored
lint (#161)
1 parent e3cff98 commit 0dbc785

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

compiler_opt/rl/policy_saver.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -220,7 +220,8 @@ def _write_output_signature(self, saver, path):
220220
}
221221
}]
222222
for info_spec in tf.nest.flatten(action_signature.info):
223-
sm_action_info = sm_action_tensors[sm_action_indices[info_spec.name.lower()]]
223+
sm_action_info = sm_action_tensors[sm_action_indices[
224+
info_spec.name.lower()]]
224225
sm_action_info = _get_non_identity_op(sm_action_info)
225226
(tensor_op, tensor_port) = _split_tensor_name(sm_action_info.name)
226227
output_list.append({

0 commit comments

Comments
 (0)